Skip to content

GitLab

Commit 3672d070 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Remove no-longer used template parameter 'NPerBlock'

parent a9bbab4d
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 10 deletions
example/36_permute/permute_fp16.cpp
View file @ 3672d070
...@@ -8,11 +8,11 @@ using BDataType = F16; ...@@ -8,11 +8,11 @@ using BDataType = F16;
// clang-format off // clang-format off
using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
// ######| InData| OutData| Elementwise| NumDim| Block| NPer| HPer| WPer| InBlock| InBlockTransfer| InBlockTransfer| Src| Dst| Src| Dst| // ######| InData| OutData| Elementwise| NumDim| Block| HPer| WPer| InBlock| InBlockTransfer| InBlockTransfer| Src| Dst| Src| Dst|
// ######| Type| Type| Operation| | Size| Block| Block| Block| LdsExtraW| ThreadClusterLengths| ThreadClusterArrangeOrder| VectorDim| VectorDim| ScalarPerVector| ScalarPerVector| // ######| Type| Type| Operation| | Size| Block| Block| LdsExtraW| ThreadClusterLengths| ThreadClusterArrangeOrder| VectorDim| VectorDim| ScalarPerVector| ScalarPerVector|
// ######| | | | | | | | | | | | | | | | // ######| | | | | | | | | | | | | | |
// ######| | | | | | | | | | | | | | | | // ######| | | | | | | | | | | | | | |
< ADataType, BDataType, PassThrough, 4, 256, 128, 128, 128, 0, S<1, 16, 16>, S<0, 1, 2>, 3, 2, 1, 1>; < ADataType, BDataType, PassThrough, 4, 256, 128, 128, 0, S<1, 16, 16>, S<0, 1, 2>, 3, 2, 1, 1>;
// clang-format on // clang-format on
#include "run_permute_example.inc" #include "run_permute_example.inc"
......
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @ 3672d070
...@@ -79,7 +79,6 @@ template <typename InDataType, ...@@ -79,7 +79,6 @@ template <typename InDataType,
typename ElementwiseOperation, typename ElementwiseOperation,
index_t NumDim, index_t NumDim,
index_t BlockSize, index_t BlockSize,
index_t NPerBlock,
index_t HPerBlock, index_t HPerBlock,
index_t WPerBlock, index_t WPerBlock,
index_t InBlockLdsExtraW, index_t InBlockLdsExtraW,
...@@ -95,7 +94,6 @@ struct DevicePermute ...@@ -95,7 +94,6 @@ struct DevicePermute
ElementwiseOperation, ElementwiseOperation,
NumDim, NumDim,
BlockSize, BlockSize,
NPerBlock,
HPerBlock, HPerBlock,
WPerBlock, WPerBlock,
InBlockLdsExtraW, InBlockLdsExtraW,
...@@ -143,7 +141,7 @@ struct DevicePermute ...@@ -143,7 +141,7 @@ struct DevicePermute
make_tuple(Sequence<0>{}, Sequence<1>{}, Sequence<2>{})); make_tuple(Sequence<0>{}, Sequence<1>{}, Sequence<2>{}));
return PadTensorDescriptor( return PadTensorDescriptor(
desc_n_h_w, make_tuple(NPerBlock, HPerBlock, WPerBlock), Sequence<true, true, true>{}); desc_n_h_w, make_tuple(1, HPerBlock, WPerBlock), Sequence<false, true, true>{});
} }
using InGridDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1})); using InGridDesc = decltype(MakeDescriptor_N_H_W({1, 1}, {1, 1}));
...@@ -156,7 +154,6 @@ struct DevicePermute ...@@ -156,7 +154,6 @@ struct DevicePermute
OutDataType, OutDataType,
ElementwiseOperation, ElementwiseOperation,
BlockSize, BlockSize,
NPerBlock,
HPerBlock, HPerBlock,
WPerBlock, WPerBlock,
InBlockLdsExtraW, InBlockLdsExtraW,
......
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
View file @ 3672d070
...@@ -98,7 +98,6 @@ template <typename InGridDesc, ...@@ -98,7 +98,6 @@ template <typename InGridDesc,
typename OutDataType, typename OutDataType,
typename ElementwiseOperation, typename ElementwiseOperation,
index_t BlockSize, index_t BlockSize,
index_t NPerBlock,
index_t HPerBlock, index_t HPerBlock,
index_t WPerBlock, index_t WPerBlock,
index_t InBlockLdsExtraW, index_t InBlockLdsExtraW,
......
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