提交 23478a0d 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

test(dnn/cuda): fix cuda int8 test on sm60

GitOrigin-RevId: 66bab333e18a32c1f06a5c2dfd6a01c368b3f674
上级 cec7fd3c
......@@ -582,7 +582,7 @@ TEST_F(CUDA, CONV_BIAS_FORWARD_MATMUL_1x1) {
TEST_F(CUDA, CONV_BIAS_FORWARD_GROUP) {
using NLMode = ConvBias::Param::NonlineMode;
bool is_int_available = false;
if (megdnn::test::check_compute_capability((6), (0))) {
if (megdnn::test::check_compute_capability(6, 1)) {
is_int_available = true;
} else {
is_int_available = false;
......
......@@ -129,7 +129,7 @@ TEST_F(CUDA, PEAK_BENCHMARK_MATRIX_MUL_QUANTIZED4x4x32) {
#endif
TEST_F(CUDA, MATRIX_MUL_INT8x8x32_WITH_SPETIAL_STRIDES) {
if (cuda::current_device_prop().major < 6) {
if (!cuda::is_compute_capability_required(6, 1)) {
printf("Skip CUDA.MATRIX_MUL test as current device doesn't support\n");
return;
}
......@@ -151,7 +151,7 @@ TEST_F(CUDA, MATRIX_MUL_INT8x8x32_WITH_SPETIAL_STRIDES) {
}
TEST_F(CUDA, MATRIX_MUL_INT8x8x32_NAIVE) {
if (cuda::current_device_prop().major < 6) {
if (!cuda::is_compute_capability_required(6, 1)) {
printf("Skip CUDA.MATRIX_MUL test as current device doesn't support\n");
return;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册