|
@@ -17,7 +17,6 @@ |
|
|
#include "test/common/convolution.h" |
|
|
#include "test/common/convolution.h" |
|
|
#include "test/common/rng.h" |
|
|
#include "test/common/rng.h" |
|
|
#include "test/cuda/benchmark.h" |
|
|
#include "test/cuda/benchmark.h" |
|
|
#include "test/cuda/utils.h" |
|
|
|
|
|
|
|
|
|
|
|
#include "src/cuda/utils.h" |
|
|
#include "src/cuda/utils.h" |
|
|
|
|
|
|
|
@@ -204,7 +203,7 @@ TEST_F(CUDA, CONVOLUTION_BACKWARD_DATA) |
|
|
.set_epsilon(1e-3) |
|
|
.set_epsilon(1e-3) |
|
|
.set_param(arg.param) |
|
|
.set_param(arg.param) |
|
|
.exec(TensorLayoutArray{filter, dst, src}); |
|
|
.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(); |
|
|
src.dtype = dst.dtype = filter.dtype = dtype::Float16(); |
|
|
checker.set_rng(0, &rng) |
|
|
checker.set_rng(0, &rng) |
|
|
.set_rng(1, &rng) |
|
|
.set_rng(1, &rng) |
|
|