From 83e4c9d7ab69643f6257ae095e05edd01b6973d7 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 5 Jul 2021 14:12:39 +0800 Subject: [PATCH] fix(opencl): open opencl topk test when opencl beyond 2.0 GitOrigin-RevId: f2ad6b4af2ffacdf9fdd60fc20cf9cff3565612c --- dnn/test/common/topk.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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 } -- GitLab