MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_args_for_cuda

Function get_args_for_cuda

dnn/test/common/local.h:30–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28};
29
30static 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
45static inline std::vector<TestArg> get_args_for_intel_gpu() {
46 std::vector<TestArg> test_args;

Callers 2

TEST_FFunction · 0.85
TEST_FFunction · 0.85

Calls 1

emplace_backMethod · 0.80

Tested by

no test coverage detected