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

added GridwiseConvolutionImplicitGemm_v1r2_nchw_cyxk_khwn

Browse Source
This commit is contained in:
Chao Liu
2019-04-19 14:22:02 -05:00
parent 19f17df47a
commit 5ce19234a4
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