Convert Python list to InfiniCore Tensor

Author: zhuyue

python/infinicore/__init__.py

@@ -26,6 +26,9 @@
     long,
     short,
     uint8,
+    uint16,
+    uint32,
+    uint64,
 )
 from infinicore.ops.add import add
 from infinicore.ops.attention import attention
@@ -41,6 +44,7 @@
     ones,
     strided_empty,
     strided_from_blob,
+    tensor,
     zeros,
 )
 
@@ -74,6 +78,9 @@
     "long",
     "short",
     "uint8",
+    "uint16",
+    "uint32",
+    "uint64",
     # Operations.
     "add",
     "attention",
@@ -87,6 +94,7 @@
     "ones",
     "strided_empty",
     "strided_from_blob",
+    "tensor",
     "zeros",
 ]
 

python/infinicore/dtype.py

@@ -68,6 +68,9 @@ def __hash__(self):
 half = float16
 bfloat16 = dtype(_infinicore.DataType.BF16)
 uint8 = dtype(_infinicore.DataType.U8)
+uint16 = dtype(_infinicore.DataType.U16)
+uint32 = dtype(_infinicore.DataType.U32)
+uint64 = dtype(_infinicore.DataType.U64)
 int8 = dtype(_infinicore.DataType.I8)
 int16 = dtype(_infinicore.DataType.I16)
 short = int16