llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

pull/3222/head
Yuanqiang Liu 2024-04-24 11:49:26 +08:00 committed by GitHub
parent 8a1dbbd597
commit dc470e65c8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/torch-mlir/Dialect/Torch/IR/TorchTypes.td
View File

@ -129,6 +129,7 @@ class AnyTorchTensorType<string name, string typeMnemonic>
| torch.bool | i1 | | torch.bool | i1 |
| torch.qint8 | !torch.qint8 | | torch.qint8 | !torch.qint8 |
| torch.quint8 | !torch.quint8 | | torch.quint8 | !torch.quint8 |
| torch.qint32 | !torch.qint32 |
| torch.complex64 | complex<f32> | | torch.complex64 | complex<f32> |
| torch.complex128 | complex<f64> | | torch.complex128 | complex<f64> |
|-------------------|--------------------| |-------------------|--------------------|