diff --git a/dnn/test/common/topk.cpp b/dnn/test/common/topk.cpp index e785c3a9..015a4979 100644 --- a/dnn/test/common/topk.cpp +++ b/dnn/test/common/topk.cpp @@ -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 }