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

[MLIR][TORCH] Add support for unsigned integer types 

Refer: https://github.com/pytorch/pytorch/issues/58734
pull/2590/head
Vivek Khandelwal 2023-11-21 15:36:34 +00:00
parent 03e8f99730
commit d50d3aa5e7
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/Dialect/Torch/IR/TorchTypes.cpp
View File

@ -200,7 +200,10 @@ static bool isValidTorchDtype(Type dtype) {
} }
} }
if (type.isUnsigned()) { if (type.isUnsigned()) {
return type.getWidth() == 8 || type.getWidth() == 4; for (unsigned width : {4, 8, 16, 32, 64}) {
if (type.getWidth() == width)
return true;
}
} }
} }
return false; return false;