diff --git a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp index 4f4d7784536c7..9e365d9202011 100644 --- a/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp +++ b/compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils/ConfigUtils.cpp @@ -164,13 +164,9 @@ static std::optional 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 unaligned schedule. Currently we + // dont have this for TileAndFuse path, see + // https://github.com/iree-org/iree/issues/19532 return schedule; }