change cuf

Author: SusanTan

flang/include/flang/Optimizer/Dialect/CUF/CUFOps.td

@@ -21,13 +21,14 @@ include "flang/Optimizer/Dialect/FIRAttr.td"
 include "mlir/Dialect/GPU/IR/GPUBase.td"
 include "mlir/Dialect/LLVMIR/LLVMOpBase.td"
 include "mlir/Interfaces/LoopLikeInterface.td"
+include "mlir/Interfaces/SideEffectInterfaces.td"
 include "mlir/IR/BuiltinAttributes.td"
 
 class cuf_Op<string mnemonic, list<Trait> traits>
     : Op<CUFDialect, mnemonic, traits>;
 
 def cuf_AllocOp : cuf_Op<"alloc", [AttrSizedOperandSegments,
-    MemoryEffects<[MemAlloc]>]> {
+    MemoryEffectsOpInterface]> {
   let summary = "Allocate an object on device";
 
   let description = [{
@@ -62,6 +63,11 @@ def cuf_AllocOp : cuf_Op<"alloc", [AttrSizedOperandSegments,
       CArg<"mlir::ValueRange", "{}">:$shape,
       CArg<"llvm::ArrayRef<mlir::NamedAttribute>", "{}">:$attributes)>];
 
+  let extraClassDeclaration = [{
+    void getEffects(
+        llvm::SmallVectorImpl<mlir::MemoryEffects::EffectInstance> &effects);
+  }];
+
   let hasVerifier = 1;
 }
 

flang/lib/Optimizer/Dialect/CUF/CUFOps.cpp

@@ -24,6 +24,7 @@
 #include "mlir/IR/Matchers.h"
 #include "mlir/IR/OpDefinition.h"
 #include "mlir/IR/PatternMatch.h"
+#include "mlir/Interfaces/SideEffectInterfaces.h"
 #include "llvm/ADT/SmallVector.h"
 
 //===----------------------------------------------------------------------===//
@@ -66,6 +67,14 @@ static llvm::LogicalResult checkCudaAttr(Op op) {
 
 llvm::LogicalResult cuf::AllocOp::verify() { return checkCudaAttr(*this); }
 
+// Attach value-scoped MemAlloc to the result value.
+void cuf::AllocOp::getEffects(
+    llvm::SmallVectorImpl<mlir::MemoryEffects::EffectInstance> &effects) {
+  effects.emplace_back(mlir::MemoryEffects::Allocate::get(),
+                       mlir::cast<mlir::OpResult>(getPtr()),
+                       mlir::SideEffects::DefaultResource::get());
+}
+
 //===----------------------------------------------------------------------===//
 // FreeOp
 //===----------------------------------------------------------------------===//