提交 83e4c9d7 编写于 作者: M Megvii Engine Team 提交者: huangxinda

fix(opencl): open opencl topk test when opencl beyond 2.0

GitOrigin-RevId: f2ad6b4af2ffacdf9fdd60fc20cf9cff3565612c
上级 f30c0e06
......@@ -170,9 +170,8 @@ void test::run_topk_test(Handle* handle) {
run(-5, 123, 3, mode); // equiv to rev sort
run(5, 3, 1231, mode, 2000); // non contig
//! opencl on armv7's CI does not support large batch.
//! but P30 and MI9 are ok. fix it in the future.
#if !defined(MEGDNN_ARMV7) && defined(MGB_CUDA)
//! opencl does not support large batch. fix it in the future.
#if MGB_CUDA
run(3, 70000, 5, mode, 10); // non contig
#endif
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册