From 0c88a3d8910d0d08b7bb775406fffc743cfa00a4 Mon Sep 17 00:00:00 2001 From: Chao Liu Date: Sat, 9 Mar 2019 13:51:08 -0600 Subject: [PATCH] update --- driver/driver.hip.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/driver/driver.hip.cpp b/driver/driver.hip.cpp index aca345acfd..2cd1ac4b24 100644 --- a/driver/driver.hip.cpp +++ b/driver/driver.hip.cpp @@ -577,10 +577,10 @@ int main(int argc, char* argv[]) ostream_ConstantTensorDescriptor(wei_kcyx_desc, std::cout << "wei_kcyx_desc: "); ostream_ConstantTensorDescriptor(out_nkhw_desc, std::cout << "out_nkhw_desc: "); - Tensor in_nchw(make_TensorDescriptor(in_nchw_desc)); - Tensor wei_kcyx(make_TensorDescriptor(wei_kcyx_desc)); - Tensor out_nkhw_host(make_TensorDescriptor(out_nkhw_desc)); - Tensor out_nkhw_device(make_TensorDescriptor(out_nkhw_desc)); + Tensor in_nchw(make_TensorDescriptor(in_nchw_desc)); + Tensor wei_kcyx(make_TensorDescriptor(wei_kcyx_desc)); + Tensor out_nkhw_host(make_TensorDescriptor(out_nkhw_desc)); + Tensor out_nkhw_device(make_TensorDescriptor(out_nkhw_desc)); std::size_t num_thread = std::thread::hardware_concurrency(); @@ -633,7 +633,7 @@ int main(int argc, char* argv[]) if(do_verification) { -#if 0 +#if 1 if(Y == 3 && X == 3) { host_winograd_3x3_convolution(in_nchw, wei_kcyx, out_nkhw_host, lower_pads, upper_pads);