From c39c573eb8607620242ab4dde4c61f94c0ed6322 Mon Sep 17 00:00:00 2001 From: Chao Liu Date: Thu, 24 Jan 2019 16:02:24 -0600 Subject: [PATCH] refactor --- .../device_implicit_gemm_convolution_2_cnhw_srck_knhw.cuh | 8 ++++---- ...ridwise_implicit_gemm_convolution_3_cnhw_srck_knhw.cuh | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/driver/device_implicit_gemm_convolution_2_cnhw_srck_knhw.cuh b/driver/device_implicit_gemm_convolution_2_cnhw_srck_knhw.cuh index 892eff2083..18c639ea1b 100644 --- a/driver/device_implicit_gemm_convolution_2_cnhw_srck_knhw.cuh +++ b/driver/device_implicit_gemm_convolution_2_cnhw_srck_knhw.cuh @@ -102,8 +102,8 @@ void device_implicit_gemm_convolution_2_cnhw_srck_knhw(InDesc, constexpr unsigned KPerThread = 8; constexpr unsigned CPerThread = 1; - constexpr unsigned GemmThreadPerClusterRow = 4; - constexpr unsigned GemmThreadPerClusterColumn = 4; + constexpr unsigned GemmRowThreadPerCluster = 4; + constexpr unsigned GemmColumnThreadPerCluster = 4; constexpr unsigned InBlockCopyThreadPerDim0 = 2; constexpr unsigned InBlockCopyThreadPerDim1 = 64; @@ -149,8 +149,8 @@ void device_implicit_gemm_convolution_2_cnhw_srck_knhw(InDesc, BPerThread, KPerThread, CPerThread, - GemmThreadPerClusterRow, - GemmThreadPerClusterColumn, + GemmRowThreadPerCluster, + GemmColumnThreadPerCluster, InBlockCopyThreadPerDim0, InBlockCopyThreadPerDim1> <<>>(in_cnhw_desc, diff --git a/src/include/gridwise_implicit_gemm_convolution_3_cnhw_srck_knhw.cuh b/src/include/gridwise_implicit_gemm_convolution_3_cnhw_srck_knhw.cuh index d0d5797790..0d1b0fca0b 100644 --- a/src/include/gridwise_implicit_gemm_convolution_3_cnhw_srck_knhw.cuh +++ b/src/include/gridwise_implicit_gemm_convolution_3_cnhw_srck_knhw.cuh @@ -20,8 +20,8 @@ template __global__ void @@ -177,8 +177,8 @@ gridwise_implicit_gemm_convolution_3_cnhw_srck_knhw(InGlobalDesc, false, false, CPerThread, - GemmThreadPerClusterRow, - GemmThreadPerClusterColumn, + GemmRowThreadPerCluster, + GemmColumnThreadPerCluster, true>{}; // LDS