IntType -> SymInt && SymInt == AnyTorchScalarType

Signed-off-by: Keshav Vinayak Jha <[email protected]>

Author: keshavvinayak01

include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td

@@ -15213,8 +15213,8 @@ def Torch_AtenScaledDotProductFlashAttentionOp : Torch_Op<"aten._scaled_dot_prod
     AnyTorchOptionalTensorType:$logsumexp,
     AnyTorchOptionalTensorType:$cum_seq_q,
     AnyTorchOptionalTensorType:$cum_seq_k,
-    Torch_IntType:$max_q,
-    Torch_IntType:$max_k,
+    Torch_SymIntType:$max_q,
+    Torch_SymIntType:$max_k,
     AnyTorchOptionalTensorType:$rng_state,
     AnyTorchOptionalTensorType:$unused,
     AnyTorchOptionalTensorType:$debug_attn_mask

projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/torch_ods_gen.py

@@ -29,7 +29,7 @@
     "int[]": "AnyTorchListOfTorchIntType",
     "int?": "AnyTorchOptionalIntType",
     "int[]?": "AnyTorchOptionalListOfTorchIntType",
-    "SymInt": "Torch_IntType",
+    "SymInt": "AnyTorchScalarType",
     "bool": "Torch_BoolType",
     "bool[]": "AnyTorchListOfTorchBoolType",
     "bool?": "AnyTorchOptionalBoolType",