ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add anther verision of batch gemm

Browse Source
This commit is contained in:
Chao Liu
2019-02-17 01:50:57 -06:00
parent 9f2e8f8bb4
commit 1cb9885058
8 changed files with 657 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
driver/driver.hip.cpp
View File

@@ -391,7 +391,7 @@ int main()
constexpr unsigned HPad = 0;
constexpr unsigned WPad = 0;
#elif 0
#elif 1
// 3x3, 34x34
constexpr unsigned N = 64;
constexpr unsigned C = 256;
@@ -593,11 +593,11 @@ int main()
device_implicit_gemm_convolution_1_nchw_kcsr_nkhw
#elif 0
device_implicit_gemm_convolution_1_nchw_srck_nkhw
#elif 0
#elif 1
device_implicit_gemm_convolution_1_chwn_csrk_khwn
#elif 0
device_implicit_gemm_convolution_2_cnhw_csrk_knhw
#elif 1
#elif 0
device_implicit_gemm_convolution_2_chwn_csrk_khwn
#endif
(in_nchw_desc, in_nchw, wei_kcsr_desc, wei_kcsr, out_nkhw_desc, out_nkhw_device, nrepeat);