"git@developer.sourcefind.cn:OpenDAS/torch-scatter.git" did not exist on "7f598ef94f8ea32b6e3df5ba7058ba8867537c34"
Commit cee2d38e authored by ltqin's avatar ltqin
Browse files

change name to PartitionedBlockwiseReduction_V2

parent 7a04ca2a
......@@ -62,22 +62,22 @@ struct BlockwiseSoftmax_V1
using ThreadClusterLengths_M_K = Sequence<MPerBlock, WaveSize / MPerXDL>;
using BlockwiseMaxReduce =
PartitionedBlockwiseReduction2<AccDataType,
BlockSize,
ThreadClusterLengths_M_K,
BlockToMKMap_M0_K_M1Adapt,
reduce::Max,
false, // param ignored
detail::AccumulateWithNanIgnore<reduce::Max, AccDataType>>;
PartitionedBlockwiseReduction_V2<AccDataType,
BlockSize,
ThreadClusterLengths_M_K,
BlockToMKMap_M0_K_M1Adapt,
reduce::Max,
false, // param ignored
detail::AccumulateWithNanIgnore<reduce::Max, AccDataType>>;
using BlockwiseSumReduce =
PartitionedBlockwiseReduction2<AccDataType,
BlockSize,
ThreadClusterLengths_M_K,
BlockToMKMap_M0_K_M1Adapt,
reduce::Add,
false, // ignored
detail::AccumulateWithNanIgnore<reduce::Add, AccDataType>>;
PartitionedBlockwiseReduction_V2<AccDataType,
BlockSize,
ThreadClusterLengths_M_K,
BlockToMKMap_M0_K_M1Adapt,
reduce::Add,
false, // ignored
detail::AccumulateWithNanIgnore<reduce::Add, AccDataType>>;
using ThreadwiseSumReduce =
ThreadwiseReduction<AccDataType,
......
......@@ -97,7 +97,7 @@ template <typename AccDataType,
bool PropagateNan,
typename Accumulation =
detail::AccumulateWithNanCheck<PropagateNan, OpReduce, AccDataType>>
struct PartitionedBlockwiseReduction2
struct PartitionedBlockwiseReduction_V2
{
static_assert(BlockSize == ThreadClusterLengths_M_K::At(0) * ThreadClusterLengths_M_K::At(1),
"The product of cluster lengths should be same as BlockSize!");
......
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