fix(megdnn): change ver 60 to use cuda capability 50

fix(megdnn): if compute capability less than 60 skip fp16 test

dnn/test/cuda/convolution.cpp

@@ -17,6 +17,7 @@
 #include "test/common/convolution.h"
 #include "test/common/rng.h"
 #include "test/cuda/benchmark.h"
+#include "test/cuda/utils.h"
 
 #include "src/cuda/utils.h"
 
@@ -203,6 +204,7 @@ TEST_F(CUDA, CONVOLUTION_BACKWARD_DATA)
                 .set_epsilon(1e-3)
                 .set_param(arg.param)
                 .exec(TensorLayoutArray{filter, dst, src});
+        if (!megdnn::test::check_compute_capability(6, 0)) {
             src.dtype = dst.dtype = filter.dtype = dtype::Float16();
             checker.set_rng(0, &rng)
                     .set_rng(1, &rng)
@@ -216,6 +218,7 @@ TEST_F(CUDA, CONVOLUTION_BACKWARD_DATA)
                     .set_param(arg.param)
                     .exec(TensorLayoutArray{filter, dst, src});
         }
+    }
 }
 
 TEST_F(CUDA, CONVOLUTION_BACKWARD_DATA_FAILED_CUDNN7_5)

src/jit/impl/halide/compiler_cuda.cpp

@@ -49,7 +49,7 @@ HalideCudaTargetTrait::FeatureSet HalideCudaTargetTrait::features(
         set.set(Target::CUDACapability32);
     } else if (in(35, 40)) {
         set.set(Target::CUDACapability35);
-    } else if (in(50, 60)) {
+    } else if (in(50, 61)) {
         set.set(Target::CUDACapability50);
     } else if (in(61, 70)) {
         set.set(Target::CUDACapability61);