Updates and Bug fixes to CUTLASS 3.3 (#1232)
This commit is contained in:
@@ -347,7 +347,7 @@ public:
|
||||
// The number of tiles for which reduction is required is either:
|
||||
// (a) the total number of output tiles (in the case of split-K)
|
||||
// (b) the number of stream-K tiles
|
||||
// To calcualte the total number of output tiles in the split-K case, we
|
||||
// To calculate the total number of output tiles in the split-K case, we
|
||||
// note that, in the split-K case, the units_per_problem_ member of Params will be
|
||||
// the total number of output tiles.
|
||||
auto reduction_tiles = params.splits_ > 1 ? params.units_per_problem_ : params.sk_tiles_;
|
||||
|
||||
Reference in New Issue
Block a user