提交 1c4a64b2 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

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

GitOrigin-RevId: 3377007b875ad1538ea5ecbac2406e6941683ab8

fix(dnn): fix include file redefined error

GitOrigin-RevId: 9d52f7ba503ce3a8df6058d450c023bef1e6d59f
上级 fd113acb
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册