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

Function TEST_F

dnn/test/cuda/matrix_mul.cpp:17–35  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

15
16#if CUDA_VERSION >= 10000
17TEST_F(CUDA, MATRIX_MUL_QUANTIZED4x4x32_EXCEPTION) {
18 if (cuda::current_device_prop().major > 7 ||
19 (cuda::current_device_prop().major == 7 &&
20 cuda::current_device_prop().minor >= 5)) {
21 printf("Skip CUDA.MATRIX_MUL_QUANTIZED4x4x32_EXCEPTION test as current "
22 "device support wmma intrinsics\n");
23 return;
24 }
25
26 Checker<MatrixMul> checker(handle_cuda(), false);
27 using Param = MatrixMul::Param;
28 Param param;
29 param.transposeB = true;
30 checker.set_param(param);
31 checker.set_dtype(0, dtype::Quantized4Asymm(1.3f, (uint8_t)3));
32 checker.set_dtype(1, dtype::Quantized4Asymm(1.3f, (uint8_t)3));
33 checker.set_dtype(2, dtype::QuantizedS32(1.3f * 1.3f));
34 ASSERT_THROW(checker.exec({{256, 256}, {256, 256}, {256, 256}}), MegDNNError);
35}
36
37TEST_F(CUDA, MATRIX_MUL_QUANTIZED4x4x32) {
38 require_compute_capability(7, 5);

Callers

nothing calls this directly

Calls 8

check_compute_capabilityFunction · 0.85
TensorLayoutClass · 0.85
set_dtypeMethod · 0.80
execMethod · 0.45
execlMethod · 0.45
execsMethod · 0.45
push_backMethod · 0.45
oprMethod · 0.45

Tested by

no test coverage detected