From 68f2e59763b504db4e7ec9debd2b2c1ba44241e0 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 26 Mar 2021 19:41:22 +0800 Subject: [PATCH] fix(mgb(ci)): fix tx1 ci testcase GitOrigin-RevId: 7ac0c4b68d9ab6f451c1d43d192d0bc0aba997e0 --- dnn/test/cuda/warp_perspective.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dnn/test/cuda/warp_perspective.cpp b/dnn/test/cuda/warp_perspective.cpp index 5c0d5547..a8d00b66 100644 --- a/dnn/test/cuda/warp_perspective.cpp +++ b/dnn/test/cuda/warp_perspective.cpp @@ -664,10 +664,10 @@ TEST_F(CUDA, WARP_PERSPECTIVE_BACKWARD_MAT_BFLOAT16) { param.imode = param::WarpPerspective::InterpolationMode::LINEAR; param.bmode = bmode; checker.set_param(param); - checker.execs({{1000, 3, 11, 12}, - {1000, 3, 3}, - {1000, 3, 10, 11}, - {1000, 3, 3}}); + checker.execs({{10, 3, 11, 12}, + {10, 3, 3}, + {10, 3, 10, 11}, + {10, 3, 3}}); } } }