[core] Fix bug in callable op interface. (#3585)

schweitzpgi · web-flow · commit 2c125c8d6aa2 · 2025-11-07T03:06:49.000Z
Signed-off-by: Eric Schweitz &lt;eschweitz@nvidia.com&gt;
diff --git a/include/cudaq/Optimizer/Dialect/Quake/QuakeOps.td b/include/cudaq/Optimizer/Dialect/Quake/QuakeOps.td
@@ -443,7 +443,9 @@ def quake_ApplyOp : QuakeOp<"apply",
 
     /// Return the callee of this operation.
     mlir::CallInterfaceCallable getCallableForCallee() {
-      return (*this)->getAttrOfType<mlir::SymbolRefAttr>(getCalleeAttrName());
+      if (getCallee())
+        return (*this)->getAttrOfType<mlir::SymbolRefAttr>(getCalleeAttrName());
+      return getIndirectCallee().front();
     }
   }];
 }

Original file line number	Diff line number	Diff line change
`@@ -443,7 +443,9 @@ def quake_ApplyOp : QuakeOp<"apply",`
`443`	`443`
`444`	`444`	`/// Return the callee of this operation.`
`445`	`445`	`mlir::CallInterfaceCallable getCallableForCallee() {`
`446`		`- return (*this)->getAttrOfType<mlir::SymbolRefAttr>(getCalleeAttrName());`
	`446`	`+ if (getCallee())`
	`447`	`+ return (*this)->getAttrOfType<mlir::SymbolRefAttr>(getCalleeAttrName());`
	`448`	`+ return getIndirectCallee().front();`
`447`	`449`	`}`
`448`	`450`	`}];`
`449`	`451`	`}`