From 82be0aaced2c9cc6b29be90323f0a4a2c35dd712 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 12 Jan 2022 15:33:47 +0800
Subject: [PATCH] test(dnn): fix compute capability requirement for NCHWX test

GitOrigin-RevId: d2f8022be112d164cb8ff1f90dbe15a4ec1bda04
---
 dnn/test/cuda/accuracy_shake.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dnn/test/cuda/accuracy_shake.cpp b/dnn/test/cuda/accuracy_shake.cpp
index 68463633..41b116b2 100644
--- a/dnn/test/cuda/accuracy_shake.cpp
+++ b/dnn/test/cuda/accuracy_shake.cpp
@@ -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};
-- 
GitLab