diff --git a/third_party/py/ml_dtypes/ml_dtypes.BUILD b/third_party/py/ml_dtypes/ml_dtypes.BUILD index 6d2bf5cdb38867..15d82868898336 100644 --- a/third_party/py/ml_dtypes/ml_dtypes.BUILD +++ b/third_party/py/ml_dtypes/ml_dtypes.BUILD @@ -38,7 +38,7 @@ pybind_extension( "_src/common.h", "_src/custom_float.h", "_src/dtypes.cc", - "_src/int4_numpy.h", + "_src/intn_numpy.h", "_src/numpy.cc", "_src/numpy.h", "_src/ufuncs.h", diff --git a/third_party/py/ml_dtypes/workspace.bzl b/third_party/py/ml_dtypes/workspace.bzl index 51505bf3a1460d..871c2890b5413e 100644 --- a/third_party/py/ml_dtypes/workspace.bzl +++ b/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 = "24084d9ed2c3d45bf83b7a9bff833aa185bf9172" - ML_DTYPES_SHA256 = "c916a3e6b3d9bdcb476f506fdbbecb6d5e9f21f82f221dfcb42b320b4e85e55a" + ML_DTYPES_COMMIT = "6f02f77c4fa624d8b467c36d1d959a9b49b07900" + ML_DTYPES_SHA256 = "c5b421a3b8549c020582b9be5e9edf8bb6e9d4284cbd44b0babe6640b4af18da" tf_http_archive( name = "ml_dtypes", build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD", diff --git a/third_party/tsl/third_party/py/ml_dtypes/ml_dtypes.BUILD b/third_party/tsl/third_party/py/ml_dtypes/ml_dtypes.BUILD index 6d2bf5cdb38867..15d82868898336 100644 --- a/third_party/tsl/third_party/py/ml_dtypes/ml_dtypes.BUILD +++ b/third_party/tsl/third_party/py/ml_dtypes/ml_dtypes.BUILD @@ -38,7 +38,7 @@ pybind_extension( "_src/common.h", "_src/custom_float.h", "_src/dtypes.cc", - "_src/int4_numpy.h", + "_src/intn_numpy.h", "_src/numpy.cc", "_src/numpy.h", "_src/ufuncs.h", diff --git a/third_party/tsl/third_party/py/ml_dtypes/workspace.bzl b/third_party/tsl/third_party/py/ml_dtypes/workspace.bzl index 51505bf3a1460d..871c2890b5413e 100644 --- a/third_party/tsl/third_party/py/ml_dtypes/workspace.bzl +++ b/third_party/tsl/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 = "24084d9ed2c3d45bf83b7a9bff833aa185bf9172" - ML_DTYPES_SHA256 = "c916a3e6b3d9bdcb476f506fdbbecb6d5e9f21f82f221dfcb42b320b4e85e55a" + ML_DTYPES_COMMIT = "6f02f77c4fa624d8b467c36d1d959a9b49b07900" + ML_DTYPES_SHA256 = "c5b421a3b8549c020582b9be5e9edf8bb6e9d4284cbd44b0babe6640b4af18da" tf_http_archive( name = "ml_dtypes", build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD",