Skip to content

Commit

Permalink
Remove upcasting schedule from TileAndFuse as that is not supported b…
Browse files Browse the repository at this point in the history
…y the pipeline

Signed-off-by: Nirvedh <nirvedh@gmail.com>
  • Loading branch information
nirvedhmeshram committed Dec 20, 2024
1 parent 7e2cdf8 commit 3bc822c
Showing 1 changed file with 3 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -164,13 +164,9 @@ static std::optional<GPUMMASchedule> getMmaScheduleFromProblemAndTarget(
problem, intrinsics, seeds, maxSharedMemoryBytes, targetSubgroupSize,
transposedLhs, transposedRhs, /*canUpcastAcc=*/false,
/*mustBeAligned*/ mustBeAligned, doCPromotion);
if (!schedule) {
// Then try again by allowing upcasting accumulator.
schedule = deduceMMASchedule(
problem, intrinsics, seeds, maxSharedMemoryBytes, targetSubgroupSize,
transposedLhs, transposedRhs, /*canUpcastAcc=*/true,
/*mustBeAligned*/ mustBeAligned, doCPromotion);
}
// TODO (nirvedhmeshram) : Add support for upcasting accumulator schedule.
// Currently we dont have this for TileAndFuse path, see
// https://github.com/iree-org/iree/issues/19532
return schedule;
}

Expand Down

0 comments on commit 3bc822c

Please sign in to comment.