Browse Source

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

GitOrigin-RevId: 3377007b87

fix(dnn): fix include file redefined error

GitOrigin-RevId: 9d52f7ba50
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
1c4a64b2af
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      dnn/test/cuda/convolution.cpp

+ 1
- 2
dnn/test/cuda/convolution.cpp View File

@@ -17,7 +17,6 @@
#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"

@@ -204,7 +203,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)) {
if (!cuda::is_compute_capability_required(6, 0)) {
src.dtype = dst.dtype = filter.dtype = dtype::Float16();
checker.set_rng(0, &rng)
.set_rng(1, &rng)


Loading…
Cancel
Save