Browse Source

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

GitOrigin-RevId: 66bab333e1
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
23478a0d53
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      dnn/test/cuda/conv_bias.cpp
  2. +2
    -2
      dnn/test/cuda/matrix_mul.cpp

+ 1
- 1
dnn/test/cuda/conv_bias.cpp View File

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


+ 2
- 2
dnn/test/cuda/matrix_mul.cpp View File

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


Loading…
Cancel
Save