This commit is contained in:
Chao Liu
2019-04-13 15:53:03 -05:00
parent 00899f191b
commit 7d8daba741
4 changed files with 45 additions and 123 deletions

View File

@@ -118,7 +118,7 @@ struct GridwiseConvolutionImplicitGemm_v1r1_chwn_cyxk_khwn
// blockwise copy
// input: format is [C, Hi, Wi, N]
const auto blockwise_in_copy =
#if 1
#if 0
Blockwise4dTensorCopy1<BlockSize,
Float,
decltype(in_chwn_global_desc),

View File

@@ -117,7 +117,7 @@ struct GridwiseConvolutionImplicitGemm_v1r2_chwn_cyxk_khwn
// blockwise copy
// input: format is [C, Hi, Wi, N]
const auto blockwise_in_copy =
#if 1
#if 0
Blockwise4dTensorCopy1<BlockSize,
Float,
decltype(in_c_h_w_n_global_desc),