From ec9de227b6374f75fa9e5e695c4e3fdca3d5ec48 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 14 Apr 2020 18:01:24 +0800
Subject: [PATCH] fix(mgb/dnn): fix opencl local opr

GitOrigin-RevId: 38e0a658a263883150d7c47ed56048947c88007f
---
 dnn/test/common/local.h | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/dnn/test/common/local.h b/dnn/test/common/local.h
index 01898c702..23fd3fbc2 100644
--- a/dnn/test/common/local.h
+++ b/dnn/test/common/local.h
@@ -68,6 +68,26 @@ static inline std::vector<TestArg> get_args_for_intel_gpu() {
     return test_args;
 }
 
+static inline std::vector<TestArg> get_args_for_opencl() {
+    std::vector<TestArg> test_args;
+
+    for (size_t N : {32, 64})
+        for (size_t IC : {1, 3, 32})
+            for (size_t OC : {1, 3, 32}) {
+                test_args.emplace_back(
+                        param::Convolution{
+                                param::Convolution::Mode::CROSS_CORRELATION, 0,
+                                0, 1, 1},
+                        N, IC, 7, 7, OC, 5, 5, 3, 3);
+                test_args.emplace_back(
+                        param::Convolution{
+                                param::Convolution::Mode::CROSS_CORRELATION, 1,
+                                1, 1, 1},
+                        N, IC, 7, 7, OC, 7, 7, 3, 3);
+            }
+    return test_args;
+}
+
 static inline std::vector<TestArg> get_args_bwd_data_for_cuda() {
     std::vector<TestArg> test_args;
     // clang-format off
-- 
GitLab