add torch.qint32 to dtype-spec in TorchTypes.td (llvm#3206)

iree-org · Apr 24, 2024 · dc470e6 · dc470e6
1 parent 8a1dbbd
commit dc470e6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/include/torch-mlir/Dialect/Torch/IR/TorchTypes.td b/include/torch-mlir/Dialect/Torch/IR/TorchTypes.td
@@ -129,6 +129,7 @@ class AnyTorchTensorType<string name, string typeMnemonic>
     | torch.bool        | i1                 |
     | torch.qint8       | !torch.qint8       |
     | torch.quint8      | !torch.quint8      |
+    | torch.qint32      | !torch.qint32      |
     | torch.complex64   | complex<f32>       |
     | torch.complex128  | complex<f64>       |
     |-------------------|--------------------|