Skip to content

Commit e6aa895

Browse files
Remove upcasting schedule from TileAndFuse as that is not supported by the pipeline
Signed-off-by: Nirvedh <nirvedh@gmail.com>
1 parent 7e2cdf8 commit e6aa895

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp

+3-7
Original file line numberDiff line numberDiff line change
@@ -164,13 +164,9 @@ static std::optional<GPUMMASchedule> getMmaScheduleFromProblemAndTarget(
164164
problem, intrinsics, seeds, maxSharedMemoryBytes, targetSubgroupSize,
165165
transposedLhs, transposedRhs, /*canUpcastAcc=*/false,
166166
/*mustBeAligned*/ mustBeAligned, doCPromotion);
167-
if (!schedule) {
168-
// Then try again by allowing upcasting accumulator.
169-
schedule = deduceMMASchedule(
170-
problem, intrinsics, seeds, maxSharedMemoryBytes, targetSubgroupSize,
171-
transposedLhs, transposedRhs, /*canUpcastAcc=*/true,
172-
/*mustBeAligned*/ mustBeAligned, doCPromotion);
173-
}
167+
// TODO (nirvedhmeshram) : Add support for unaligned schedule. Currently we
168+
// dont have this for TileAndFuse path, see
169+
// https://github.com/iree-org/iree/issues/19532
174170
return schedule;
175171
}
176172

0 commit comments

Comments
 (0)