diff --git a/mlir/include/Quantum/IR/QuantumOps.td b/mlir/include/Quantum/IR/QuantumOps.td index f1fcf50c8e..9efdd66286 100644 --- a/mlir/include/Quantum/IR/QuantumOps.td +++ b/mlir/include/Quantum/IR/QuantumOps.td @@ -82,12 +82,12 @@ def DeviceInitOp : Quantum_Op<"device"> { let arguments = (ins Optional:$shots, StrAttr:$lib, - StrAttr:$name, + StrAttr:$device_name, StrAttr:$kwargs ); let assemblyFormat = [{ - (`shots` `(` $shots^ `)`)? `[` $lib `,` $name `,` $kwargs `]` attr-dict + (`shots` `(` $shots^ `)`)? `[` $lib `,` $device_name `,` $kwargs `]` attr-dict }]; } @@ -663,11 +663,11 @@ def YieldOp : Quantum_Op<"yield", [Pure, ReturnLike, Terminator, ParentOneOf<["A let summary = "Return results from quantum program regions"; let arguments = (ins - Variadic:$results + Variadic:$retvals ); let assemblyFormat = [{ - attr-dict ($results^ `:` type($results))? + attr-dict ($retvals ^ `:` type($retvals))? }]; let builders = [ diff --git a/mlir/lib/Mitigation/Transforms/MitigationMethods/Zne.cpp b/mlir/lib/Mitigation/Transforms/MitigationMethods/Zne.cpp index 616ab66d98..7d8d06bc56 100644 --- a/mlir/lib/Mitigation/Transforms/MitigationMethods/Zne.cpp +++ b/mlir/lib/Mitigation/Transforms/MitigationMethods/Zne.cpp @@ -391,7 +391,7 @@ FlatSymbolRefAttr ZneLowering::getOrInsertFoldedCircuit(Location loc, PatternRew Operation *shots = deviceInitOp.getShots().getDefiningOp(); StringAttr lib = deviceInitOp.getLibAttr(); - StringAttr name = deviceInitOp.getNameAttr(); + StringAttr name = deviceInitOp.getDeviceNameAttr(); StringAttr kwargs = deviceInitOp.getKwargsAttr(); TypeRange originalTypes = op.getArgumentTypes(); diff --git a/mlir/lib/Quantum/Transforms/ConversionPatterns.cpp b/mlir/lib/Quantum/Transforms/ConversionPatterns.cpp index 70f5741417..ccf03b741a 100644 --- a/mlir/lib/Quantum/Transforms/ConversionPatterns.cpp +++ b/mlir/lib/Quantum/Transforms/ConversionPatterns.cpp @@ -196,7 +196,7 @@ struct DeviceInitOpPattern : public OpConversionPattern { catalyst::ensureFunctionDeclaration(rewriter, op, qirName, qirSignature); auto rtd_lib = op.getLib().str(); - auto rtd_name = op.getName().str(); + auto rtd_name = op.getDeviceName().str(); auto rtd_kwargs = op.getKwargs().str(); auto rtd_lib_gs = getGlobalString(loc, rewriter, rtd_lib,