fix: support uint dtypes in torch backend

ivy-llc · Jul 23, 2024 · cd847fd · cd847fd
1 parent 0d97a31
commit cd847fd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ivy/functional/backends/torch/data_type.py b/ivy/functional/backends/torch/data_type.py
@@ -15,6 +15,9 @@
     torch.int32: "int32",
     torch.int64: "int64",
     torch.uint8: "uint8",
+    torch.uint16: "uint16",
+    torch.uint32: "uint32",
+    torch.uint64: "uint64",
     torch.bfloat16: "bfloat16",
     torch.float16: "float16",
     torch.float32: "float32",
@@ -30,6 +33,9 @@
     "int32": torch.int32,
     "int64": torch.int64,
     "uint8": torch.uint8,
+    "uint16": torch.uint16,
+    "uint32": torch.uint32,
+    "uint64": torch.uint64,
     "bfloat16": torch.bfloat16,
     "float16": torch.float16,
     "float32": torch.float32,