ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-16 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor direct

Browse Source 
[ROCm/composable_kernel commit: 24d2f034fa]
This commit is contained in:
Chao Liu
2018-11-25 01:10:11 -06:00
parent 40ddf8c139
commit 7569eeaf55
14 changed files with 253 additions and 1291 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/include/threadwise_direct_convolution.cuh
View File

@@ -9,10 +9,10 @@ __device__ void threadwise_direct_convolution(InDesc,
OutDesc,
TFloat* __restrict__ p_out)
{
constexpr auto I0 = Index<0>{};
constexpr auto I1 = Index<1>{};
constexpr auto I2 = Index<2>{};
constexpr auto I3 = Index<3>{};
constexpr auto I0 = Number<0>{};
constexpr auto I1 = Number<1>{};
constexpr auto I2 = Number<2>{};
constexpr auto I3 = Number<3>{};
constexpr auto in_desc = InDesc{};
constexpr auto wei_desc = WeiDesc{};