fix(mgb): add compute capability check

GitOrigin-RevId: fba819afb3

src/gopt/test/inference.cpp

@@ -2727,6 +2727,9 @@ TEST(TestGoptInference, ConvertFormatNCHW4) {
 
 TEST(TestGoptInference, ConvertFormatNCHW4Ic3) {
     REQUIRE_GPU(1);
+    auto cn = CompNode::load("gpu0");
+    cn.activate();
+    REQUIRE_CUDA_COMPUTE_CAPABILITY(6, 1);
     HostTensorGenerator<dtype::Float32, RandomDistribution::UNIFORM> gen{
             1.2f, 127 * 127};
     auto graph = ComputingGraph::make();

src/tensorrt/test/opr_replace.cpp

@@ -1980,6 +1980,10 @@ TEST(TestTensorRTReplace, FuseConvAdd) {
 
 TEST(TestTensorRTReplace, FuseConvAddNchw2nchw4) {
     REQUIRE_GPU(1);
+    auto cn = CompNode::load("gpu0");
+    cn.activate();
+    REQUIRE_CUDA_COMPUTE_CAPABILITY(6, 1);
+
     HostTensorGenerator<dtype::Float32, RandomDistribution::UNIFORM> gen{
             1.2f, 127 * 127};
     auto graph = ComputingGraph::make();