Lines Matching defs:multiplicity

78 static GpuIdBuilderFnType commonLinearIdBuilderFn(int64_t multiplicity = 1) {
79   auto res = [multiplicity](RewriterBase &rewriter, Location loc,
92         rewriter, loc, d0.floorDiv(multiplicity), {linearId});
125                                multiplicity},
136 static GpuIdBuilderFnType common3DIdBuilderFn(int64_t multiplicity = 1) {
137   auto res = [multiplicity](RewriterBase &rewriter, Location loc,
145     // In the 3-D mapping case, scale the first dimension by the multiplicity.
149         rewriter, loc, d0.floorDiv(multiplicity), {scaledIds[0]}));
150     // In the 3-D mapping case, unscale the first dimension by the multiplicity.
152     forallMappingSizeInOriginalBasis[0] *= multiplicity;
191                   ? commonLinearIdBuilderFn<BlockIdOp>(/*multiplicity=*/1)
192                   : common3DIdBuilderFn<BlockIdOp>(/*multiplicity=*/1);
204                         /*multiplicity=*/kNumWarpsPerGroup * warpSize)
206                         /*multiplicity=*/kNumWarpsPerGroup * warpSize);
218           ? commonLinearIdBuilderFn<ThreadIdOp>(/*multiplicity=*/warpSize)
219           : common3DIdBuilderFn<ThreadIdOp>(/*multiplicity=*/warpSize);
227                   ? commonLinearIdBuilderFn<ThreadIdOp>(/*multiplicity=*/1)
228                   : common3DIdBuilderFn<ThreadIdOp>(/*multiplicity=*/1);