| 28 | }; |
| 29 | |
| 30 | static inline std::vector<TestArg> get_args_for_cuda() { |
| 31 | std::vector<TestArg> test_args; |
| 32 | // clang-format off |
| 33 | for (size_t N: {32, 64}) |
| 34 | for (size_t IC: {1, 3, 8, 32, 33, 65}) |
| 35 | for (size_t OC: {1, 3, 8, 32, 33, 65}) { |
| 36 | test_args.emplace_back( |
| 37 | param::Convolution{param::Convolution::Mode::CROSS_CORRELATION, |
| 38 | 0, 0, 1, 1}, |
| 39 | N, IC, 7, 7, OC, 5, 5, 3, 3); |
| 40 | } |
| 41 | // clang-format on |
| 42 | return test_args; |
| 43 | } |
| 44 | |
| 45 | static inline std::vector<TestArg> get_args_for_intel_gpu() { |
| 46 | std::vector<TestArg> test_args; |
no test coverage detected