Skip to content

GitLab

"docs_zh_CN/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "4c8bfb48ef48bafae046387f82a12a33d5c76194"
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, ...@@ -82,7 +82,6 @@ template <typename InDataType,
index_t NPerBlock, index_t NPerBlock,
index_t HPerBlock, index_t HPerBlock,
index_t WPerBlock, index_t WPerBlock,
index_t MPerThread,
index_t InScalarPerVector, index_t InScalarPerVector,
index_t OutScalarPerVector> index_t OutScalarPerVector>
struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
...@@ -93,7 +92,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, ...@@ -93,7 +92,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
NPerBlock, NPerBlock,
HPerBlock, HPerBlock,
WPerBlock, WPerBlock,
MPerThread,
InScalarPerVector, InScalarPerVector,
OutScalarPerVector>> OutScalarPerVector>>
{ {
...@@ -149,7 +147,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, ...@@ -149,7 +147,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
NPerBlock, NPerBlock,
HPerBlock, HPerBlock,
WPerBlock, WPerBlock,
MPerThread,
InScalarPerVector, InScalarPerVector,
OutScalarPerVector>; OutScalarPerVector>;
...@@ -221,11 +218,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType, ...@@ -221,11 +218,6 @@ struct DevicePermute : detail::DevicePermuteBase<DevicePermute<InDataType,
static bool IsSupportedArgument(const Argument& arg) static bool IsSupportedArgument(const Argument& arg)
{ {
if(arg.inLengths_.back() % MPerThread != 0)
{
return false;
}
// check if only swap last 2 dimensions // check if only swap last 2 dimensions
if(!(std::equal(begin(arg.inLengths_), if(!(std::equal(begin(arg.inLengths_),
std::prev(end(arg.inLengths_), 2), 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, ...@@ -123,7 +123,6 @@ template <typename InGridDesc,
index_t NPerBlock, index_t NPerBlock,
index_t HPerBlock, index_t HPerBlock,
index_t WPerBlock, index_t WPerBlock,
index_t MPerThread,
index_t InScalarPerVector, index_t InScalarPerVector,
index_t OutScalarPerVector> index_t OutScalarPerVector>
struct GridwisePermute 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