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

adding implicit gemm

Browse Source 
[ROCm/composable_kernel commit: 84d9802d30]
This commit is contained in:
Chao Liu
2019-01-15 00:11:30 -06:00
parent 50256bbcfe
commit aa885b185d
15 changed files with 268 additions and 169 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/include/threadwise_direct_convolution.cuh
View File

@@ -1,5 +1,5 @@
#pragma once
#include "constant_tensor_descriptor.cuh"
#include "ConstantTensorDescriptor.cuh"
// optimized for scenario if p_in, p_wei, p_out are in register
template <class Float, class InDesc, class WeiDesc, class OutDesc>