Skip to content

GitLab

Commit 9a06e83e authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Remove 'MPerThread' template parameter

parent 7268c739
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 9 deletions
include/ck/tensor_operation/gpu/device/device_permute.hpp
View file @ 9a06e83e
......@@ -82,7 +82,6 @@ template <typename InDataType,
index_t NPerBlock,
index_t HPerBlock,
index_t WPerBlock,
index_t MPerThread,
index_t InScalarPerVector,
index_t OutScalarPerVector>
struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
......@@ -93,7 +92,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
NPerBlock,
HPerBlock,
WPerBlock,
MPerThread,
InScalarPerVector,
OutScalarPerVector>>
{
......@@ -149,7 +147,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
NPerBlock,
HPerBlock,
WPerBlock,
MPerThread,
InScalarPerVector,
OutScalarPerVector>;
......@@ -221,11 +218,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
static bool IsSupportedArgument(const Argument& arg)
{
if(arg.inLengths_.back() % MPerThread != 0)
{
return false;
}
// check if only swap last 2 dimensions
if(!(std::equal(begin(arg.inLengths_),
std::prev(end(arg.inLengths_), 2),
......
include/ck/tensor_operation/gpu/grid/gridwise_permute.hpp
View file @ 9a06e83e
......@@ -123,7 +123,6 @@ template <typename InGridDesc,
index_t NPerBlock,
index_t HPerBlock,
index_t WPerBlock,
index_t MPerThread,
index_t InScalarPerVector,
index_t OutScalarPerVector>
struct GridwisePermute
......
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