Skip to content

GitLab

Commit 97ba755f authored by Chao Liu's avatar Chao Liu
Browse files

refactor

parent 8d460740
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
src/include/threadwise_direct_convolution.hip.hpp
View file @ 97ba755f
...@@ -80,10 +80,8 @@ __device__ void threadwise_direct_convolution_2(InDesc, ...@@ -80,10 +80,8 @@ __device__ void threadwise_direct_convolution_2(InDesc,
constexpr auto wei_desc = WeiDesc{}; constexpr auto wei_desc = WeiDesc{};
constexpr auto out_desc = OutDesc{}; constexpr auto out_desc = OutDesc{};
constexpr auto in_reg_desc = constexpr auto in_reg_desc = make_ConstantTensorDescriptor_packed(in_desc.GetLengths());
make_ConstantTensorDescriptor_default_rank_packed(in_desc.GetLengths()); constexpr auto wei_reg_desc = make_ConstantTensorDescriptor_packed(wei_desc.GetLengths());
constexpr auto wei_reg_desc =
make_ConstantTensorDescriptor_default_rank_packed(wei_desc.GetLengths());
// register // register
TInWei p_in_reg[in_reg_desc.GetElementSpace()]; TInWei p_in_reg[in_reg_desc.GetElementSpace()];
......
src/include/threadwise_tensor_slice_op.hip.hpp
View file @ 97ba755f
...@@ -19,7 +19,7 @@ __device__ void threadwise_tensor_slice_copy(SrcDesc, ...@@ -19,7 +19,7 @@ __device__ void threadwise_tensor_slice_copy(SrcDesc,
constexpr auto src_desc = SrcDesc{}; constexpr auto src_desc = SrcDesc{};
constexpr auto dst_desc = DstDesc{}; constexpr auto dst_desc = DstDesc{};
constexpr auto ref_desc = make_ConstantTensorDescriptor_default_rank_packed(SrcOpLengths{}); constexpr auto ref_desc = make_ConstantTensorDescriptor_packed(SrcOpLengths{});
#if 0 #if 0
if(get_thread_local_1d_id() == 0 && get_block_1d_id() == 0) if(get_thread_local_1d_id() == 0 && get_block_1d_id() == 0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment