[XLA] Add S1 and U1 as data types

PiperOrigin-RevId: 707136925

Author: amitsabne1

third_party/py/ml_dtypes/workspace.bzl

@@ -7,8 +7,8 @@ float8 varieties, and int4.
 load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 
 def repo():
-    ML_DTYPES_COMMIT = "c12281a501469d553483eb4d68065826b9c2fcb5"
-    ML_DTYPES_SHA256 = "cee11c4bed5147bece9e385a88c20887344ad9b89b3acb09bf3d7c9c21fb9715"
+    ML_DTYPES_COMMIT = "215c9f02a121e6286662b2efd30546c71054d5e5"
+    ML_DTYPES_SHA256 = "4a03237ef6345e1467a33d126176b9c6a7539b0f60a34b344f39b3c9e8b82438"
     tf_http_archive(
         name = "ml_dtypes",
         build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD",

tsl/platform/ml_dtypes.h

@@ -28,6 +28,8 @@ using float8_e4m3b11fnuz = ::ml_dtypes::float8_e4m3b11fnuz;
 using float8_e5m2 = ::ml_dtypes::float8_e5m2;
 using float8_e5m2fnuz = ::ml_dtypes::float8_e5m2fnuz;
 
+using int1 = ::ml_dtypes::int1;
+using uint1 = ::ml_dtypes::uint1;
 using int2 = ::ml_dtypes::int2;
 using uint2 = ::ml_dtypes::uint2;
 using int4 = ::ml_dtypes::int4;