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

Remove redundant parameter in helper lambda function

parent 87d41af6
......@@ -232,26 +232,26 @@ struct DevicePermute
static bool IsSupportedArgument(const Argument& arg)
{
constexpr auto IsScalarPerVectorValid = [](const std::array<index_t, NumDim>& lengths,
const std::array<index_t, NumDim>& strides,
index_t vectorDim,
index_t scalarPerVector) {
if(strides[vectorDim] == 1 && lengths[vectorDim] % scalarPerVector == 0)
{
return true;
}
else if(strides[vectorDim] != 1 && scalarPerVector == 1)
{
return true;
}
return false;
};
return IsScalarPerVectorValid(
arg.inLengths_, arg.inStrides_, SrcVectorDim, SrcScalarPerVector) &&
IsScalarPerVectorValid(
arg.outLengths_, arg.outStrides_, DstVectorDim, DstScalarPerVector) &&
constexpr auto IsScalarPerVectorValid =
[](index_t length, index_t stride, index_t scalarPerVector) {
if(stride == 1 && length % scalarPerVector == 0)
{
return true;
}
else if(stride != 1 && scalarPerVector == 1)
{
return true;
}
return false;
};
return IsScalarPerVectorValid(arg.inLengths_[SrcVectorDim],
arg.inStrides_[SrcVectorDim],
SrcScalarPerVector) &&
IsScalarPerVectorValid(arg.outLengths_[DstVectorDim],
arg.outStrides_[DstVectorDim],
DstScalarPerVector) &&
GridwisePermute::CheckValidity(arg.in_grid_desc_, arg.out_grid_desc_);
};
};
......
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