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

Refer: pytorch/pytorch#58734
llvm · Nov 21, 2023 · d50d3aa · d50d3aa
1 parent 03e8f99
commit d50d3aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/Dialect/Torch/IR/TorchTypes.cpp b/lib/Dialect/Torch/IR/TorchTypes.cpp
@@ -200,7 +200,10 @@ static bool isValidTorchDtype(Type dtype) {
       }
     }
     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;