You need to sign in or sign up before continuing.
提交 ec9de227 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

fix(mgb/dnn): fix opencl local opr

GitOrigin-RevId: 38e0a658a263883150d7c47ed56048947c88007f
上级 f5833a52
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册