ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-18 03:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding implicit gemm v3

Browse Source 
[ROCm/composable_kernel commit: 1cc683a3a3]
This commit is contained in:
Chao Liu
2019-05-23 22:10:40 -05:00
parent 979dc4da2e
commit 39b829b919
16 changed files with 347 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/include/threadwise_direct_convolution.hip.hpp
View File

@@ -80,8 +80,10 @@ __device__ void threadwise_direct_convolution_2(InDesc,
constexpr auto wei_desc = WeiDesc{};
constexpr auto out_desc = OutDesc{};
constexpr auto in_reg_desc = make_ConstantTensorDescriptor(in_desc.GetLengths());
constexpr auto wei_reg_desc = make_ConstantTensorDescriptor(wei_desc.GetLengths());
constexpr auto in_reg_desc =
make_ConstantTensorDescriptor_default_rank_packed(in_desc.GetLengths());
constexpr auto wei_reg_desc =
make_ConstantTensorDescriptor_default_rank_packed(wei_desc.GetLengths());
// register
TInWei p_in_reg[in_reg_desc.GetElementSpace()];