Follow upstream changes on naming

Author: eymay

lib/Dialect/BGV/IR/BGVOps.td

@@ -107,7 +107,7 @@ def BGV_RotateColumnsOp : BGV_Op<"rotate_cols", [AllTypesMatch<["input", "output
 
   let arguments = (ins
     LWECiphertext:$input,
-    Builtin_IntegerAttr:$offset
+    Builtin_IntegerAttr:$static_shift
   );
 
   let results = (outs
@@ -133,7 +133,7 @@ def BGV_RotateRowsOp : BGV_Op<"rotate_rows", [AllTypesMatch<["input", "output"]>
 
   let arguments = (ins
     LWECiphertext:$input,
-    Builtin_IntegerAttr:$offset
+    Builtin_IntegerAttr:$static_shift
   );
 
   let results = (outs

lib/Dialect/CKKS/IR/CKKSOps.td

@@ -97,7 +97,7 @@ def CKKS_RotateOp : CKKS_Op<"rotate", [AllTypesMatch<["input", "output"]>]> {
 
   let arguments = (ins
     LWECiphertext:$input,
-    Builtin_IntegerAttr:$offset
+    Builtin_IntegerAttr:$static_shift
   );
 
   let results = (outs
@@ -173,9 +173,6 @@ def CKKS_RelinearizeOp : CKKS_Op<"relinearize", [SameOperandsAndResultRings,
   let results = (outs
     LWECiphertext:$output
   );
-
-  // let hasVerifier = 1;
-  let assemblyFormat = "operands attr-dict `:` qualified(type($input)) `->` qualified(type($output))" ;
 }
 
 def CKKS_RescaleOp : CKKS_Op<"rescale"> {

lib/Dialect/LWE/Conversions/LWEToOpenfhe/LWEToOpenfhe.h

@@ -114,9 +114,11 @@ struct ConvertRotateOp : public OpConversionPattern<RotateOp> {
     Value cryptoContext = result.value();
     Location loc = op.getLoc();
 
-    // Get the rotation index
-    auto rotIndexAttr = adaptor.getOffset();
-    int64_t rotIndex = rotIndexAttr.getInt();
+    // Get the rotation index, normalized to i64 so the kmrt key type and the
+    // arith.constant agree regardless of whether the input attr was i32/i64.
+    auto rotIndexAttrIn = adaptor.getStaticShift();
+    int64_t rotIndex = rotIndexAttrIn.getInt();
+    auto rotIndexAttr = rewriter.getI64IntegerAttr(rotIndex);
 
     // Create a constant for the rotation index
     Value rotIndexValue = rewriter.create<mlir::arith::ConstantOp>(