Correct type conversions in OpBuilder and add corresponding test. (#8873)

Author: iwknow

test/test_operations.py

@@ -2632,6 +2632,11 @@ def aten_fn(b, A, lower, unit_diagonal, transpose_a):
             'transpose_a': False
         })
 
+  def test_type_conversion(self):
+    for xla_type in xb._XLA_PT_TYPE_MAP:
+      pt_type = xb.Op.to_torch_type(xla_type)
+      self.assertEqual(xla_type, xb.Op.from_torch_type(pt_type))
+
 
 class MpDecoratorTest(test_utils.XlaTestCase):
 

torch_xla/core/xla_builder.py

@@ -29,11 +29,11 @@ class Type:
     Type.F16: torch.float16,
     Type.U8: torch.uint8,
     Type.S8: torch.int8,
-    Type.U16: torch.int16,
+    Type.U16: torch.uint16,
     Type.S16: torch.int16,
-    Type.U32: torch.int32,
+    Type.U32: torch.uint32,
     Type.S32: torch.int32,
-    Type.U64: torch.int64,
+    Type.U64: torch.uint64,
     Type.S64: torch.int64,
     Type.C64: torch.complex64,
     Type.C128: torch.complex128,