提交 82be0aac 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: d2f8022be112d164cb8ff1f90dbe15a4ec1bda04
上级 619d78ed
...@@ -97,7 +97,7 @@ TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NHWC) { ...@@ -97,7 +97,7 @@ TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NHWC) {
TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NCHWX) { TEST_F(CUDA, SHAKE_CONV_BIAS_FORWARD_QS8_NCHWX) {
using Format = ConvBias::Param::Format; using Format = ConvBias::Param::Format;
require_compute_capability(6, 1); require_compute_capability(7, 5);
AccuracyShakeChecker<ConvBiasForward> checker(handle_cuda()); AccuracyShakeChecker<ConvBiasForward> checker(handle_cuda());
UniformIntRNG int_rng{-5, 5}; UniformIntRNG int_rng{-5, 5};
UniformFloatRNG float_rng{-50, 50}; UniformFloatRNG float_rng{-50, 50};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册