Browse Source

fix(mgb/dnn): fix opencl local opr

GitOrigin-RevId: 38e0a658a2
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
ec9de227b6
1 changed files with 20 additions and 0 deletions
  1. +20
    -0
      dnn/test/common/local.h

+ 20
- 0
dnn/test/common/local.h View File

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


Loading…
Cancel
Save