Update llvm version for pack propagation and linalg memory effects helpers (#136)

Author: Longsheng Du

cmake/llvm-version.txt

@@ -1 +1 @@
-7042fcc6389c6c103d501b6f39988eafed0d9b5b
+37661a17e26d9002ae9ade8c0de3932c22f16360

lib/gc/Dialect/Linalgx/LinalgOps.cpp.inc

@@ -525,20 +525,25 @@ private:
 static void getGenericEffectsImpl(
     SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>
         &effects,
-    ValueRange results, const ValueRange inputOperands,
-    ValueRange outputOperands) {
-  for (auto operand : inputOperands) {
+    LinalgOp linalgOp) {
+  for (auto [index, operand] : llvm::enumerate(linalgOp.getDpsInputs())) {
     if (!llvm::isa<MemRefType>(operand.getType()))
       continue;
-    effects.emplace_back(MemoryEffects::Read::get(), operand,
-                         SideEffects::DefaultResource::get());
+    effects.emplace_back(
+        MemoryEffects::Read::get(), &linalgOp->getOpOperand(index), /*stage=*/0,
+        /*effectOnFullRegion=*/true, SideEffects::DefaultResource::get());
   }
-  for (auto operand : outputOperands) {
-    if (!llvm::isa<MemRefType>(operand.getType()))
+
+  for (OpOperand &operand : linalgOp.getDpsInitsMutable()) {
+    if (!llvm::isa<MemRefType>(operand.get().getType()))
       continue;
-    effects.emplace_back(MemoryEffects::Read::get(), operand,
-                         SideEffects::DefaultResource::get());
-    effects.emplace_back(MemoryEffects::Write::get(), operand,
+    if (linalgOp.payloadUsesValueFromOperand(&operand)) {
+      effects.emplace_back(MemoryEffects::Read::get(), &operand, /*stage=*/0,
+                           /*effectOnFullRegion=*/true,
+                           SideEffects::DefaultResource::get());
+    }
+    effects.emplace_back(MemoryEffects::Write::get(), &operand, /*stage=*/0,
+                         /*effectOnFullRegion=*/true,
                          SideEffects::DefaultResource::get());
   }
 }

lib/gc/Dialect/Linalgx/LinalgxOps.cpp

@@ -78,8 +78,7 @@ void SigmoidOp::getEffects(
         &effects) {
   if (hasPureTensorSemantics())
     return;
-  getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
-                        getDpsInits());
+  getGenericEffectsImpl(effects, cast<LinalgOp>(getOperation()));
 }
 
 //===----------------------------------------------------------------------===//
@@ -188,8 +187,7 @@ void Mm2DVnniOp::getEffects(
         &effects) {
   if (hasPureTensorSemantics())
     return;
-  getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
-                        getDpsInits());
+  getGenericEffectsImpl(effects, cast<LinalgOp>(getOperation()));
 }
 
 LogicalResult Mm2DVnniOp::verify() {
@@ -329,8 +327,7 @@ void Mm4DVnniOp::getEffects(
         &effects) {
   if (hasPureTensorSemantics())
     return;
-  getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
-                        getDpsInits());
+  getGenericEffectsImpl(effects, cast<LinalgOp>(getOperation()));
 }
 
 LogicalResult Mm4DVnniOp::verify() {
@@ -472,8 +469,7 @@ void BatchReduceMatmulVnniOp::getEffects(
         &effects) {
   if (hasPureTensorSemantics())
     return;
-  getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
-                        getDpsInits());
+  getGenericEffectsImpl(effects, cast<LinalgOp>(getOperation()));
 }
 
 LogicalResult BatchReduceMatmulVnniOp::verify() {
@@ -609,8 +605,7 @@ void MultiBatchMatmulOp::getEffects(
         &effects) {
   if (hasPureTensorSemantics())
     return;
-  getGenericEffectsImpl(effects, getOperation()->getResults(), getDpsInputs(),
-                        getDpsInits());
+  getGenericEffectsImpl(effects, cast<LinalgOp>(getOperation()));
 }
 
 /////// Operations corresponding to library calls defined with Tablegen ////////