ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-13 01:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

initial cuda run

Browse Source
This commit is contained in:
Chao Liu
2018-10-22 20:51:26 -05:00
parent 2f2cf35bf4
commit 49ceb0fe04
2 changed files with 15 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/include/direct_convolution.cuh
View File

@@ -3,10 +3,10 @@
template <class TFloat, int NBlockDim>
__global__ void direct_convolution(DeviceTensorDescriptor in_desc,
TFloat* const in,
TFloat* const p_in,
DeviceTensorDescriptor wei_desc,
TFloat* const wei,
TFloat* const p_wei,
DeviceTensorDescriptor out_desc,
TFloat* out)
TFloat* p_out)
{
}