refactor

97ba755f · Chao Liu · 8d460740 · 97ba755f · 97ba755f
Commit 97ba755f authored May 31, 2019 by Chao Liu
Showing with 3 additions and 5 deletions

src/include/threadwise_direct_convolution.hip.hpp src/include/threadwise_direct_convolution.hip.hpp +2 -4

src/include/threadwise_tensor_slice_op.hip.hpp src/include/threadwise_tensor_slice_op.hip.hpp +1 -1

No files found.
--- a/src/include/threadwise_direct_convolution.hip.hpp
+++ b/src/include/threadwise_direct_convolution.hip.hpp
@@ -80,10 +80,8 @@ __device__ void threadwise_direct_convolution_2(InDesc,
    constexpr auto wei_desc = WeiDesc{};
    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
    TInWei p_in_reg[in_reg_desc.GetElementSpace()];

--- a/src/include/threadwise_tensor_slice_op.hip.hpp
+++ b/src/include/threadwise_tensor_slice_op.hip.hpp
@@ -19,7 +19,7 @@ __device__ void threadwise_tensor_slice_copy(SrcDesc,
    constexpr auto src_desc = SrcDesc{};
    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(get_thread_local_1d_id() == 0 && get_block_1d_id() == 0)