Fix correctness issue

Remove check from collapse dimensions to paper over correctness issue
iree-org · Sep 8, 2024 · 3564b56 · 3564b56
1 parent 60edb94
commit 3564b56
Showing 1 changed file with 0 additions and 17 deletions.
diff --git a/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp b/compiler/src/iree/compiler/DispatchCreation/CollapseDimensions.cpp
@@ -170,23 +170,6 @@ static bool isEligibleForCollapse(linalg::GenericOp genericOp) {
   if (genericOp.hasIndexSemantics())
     return false;
 
-  // TODO(#17948) GPU codegen fails when we collapse the dimensions of softmax.
-  if (llvm::any_of(genericOp.getDpsInputOperands(),
-                   [&](OpOperand *operand) -> bool {
-                     auto genericOperand =
-                         operand->get().getDefiningOp<linalg::GenericOp>();
-                     if (!genericOperand)
-                       return false;
-
-                     if (genericOperand.getNumReductionLoops() == 0)
-                       return false;
-
-                     return genericOp.getMatchingIndexingMap(operand)
-                         .isProjectedPermutation();
-                   })) {
-    return false;
-  }
-
   return true;
 }