[ROCm/composable_kernel commit: 701b7341f0]
This commit is contained in:
Chao Liu
2019-08-07 19:25:54 -05:00
parent 139b043d07
commit 0b8efbb3e9
4 changed files with 38 additions and 199 deletions

View File

@@ -379,7 +379,7 @@ int main(int argc, char* argv[])
#elif 0
device_convolution_implicit_gemm_v3_nchw_cyxk_nkhw(
(in_nchw_desc, in_nchw, wei_kcyx_desc, wei_kcyx, out_nkhw_desc, out_nkhw_device, nrepeat);
#elif 0
#elif 1
device_convolution_implicit_gemm_v4r1_nchw_kcyx_nkhw(in_nchw_desc,
in_nchw,
wei_kcyx_desc,