Browse Source

test(dnn): fix compute capability requirement for NCHWX test

GitOrigin-RevId: d2f8022be1
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
82be0aaced
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      dnn/test/cuda/accuracy_shake.cpp

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

@@ -97,7 +97,7 @@ TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NHWC) {

TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NCHWX) {
using Format = ConvBias::Param::Format;
require_compute_capability(6, 1);
require_compute_capability(7, 5);
AccuracyShakeChecker<ConvBiasForward> checker(handle_cuda());
UniformIntRNG int_rng{-5, 5};
UniformFloatRNG float_rng{-50, 50};


Loading…
Cancel
Save