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

added GridwiseConvolutionImplicitGemm_v1r2_nchw_cyxk_khwn

Browse Source 
[ROCm/composable_kernel commit: 5ce19234a4]
This commit is contained in:
Chao Liu
2019-04-19 14:22:02 -05:00
parent d0244d3a51
commit f367814ef7
7 changed files with 135 additions and 406 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
driver/driver.hip.cpp
View File

@@ -673,7 +673,7 @@ int main(int argc, char* argv[])
device_direct_convolution_2_nchw_kcyx_nkhw
#elif 0
device_direct_convolution_2_vectorized_nchw_kcyx_nkhw
#elif 1
#elif 0
device_convolution_implicit_gemm_v1_chwn_cyxk_khwn
#elif 1
device_convolution_implicit_gemm_v1_nchw_cyxk_khwn