Browse Source

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

GitOrigin-RevId: f2ad6b4af2
release-1.5
Megvii Engine Team huangxinda 4 years ago
parent
commit
83e4c9d7ab
1 changed files with 2 additions and 3 deletions
  1. +2
    -3
      dnn/test/common/topk.cpp

+ 2
- 3
dnn/test/common/topk.cpp View File

@@ -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
}


Loading…
Cancel
Save