Skip to content

Commit 01d2716

Browse files
committed
[TSL] Bump ml_dtypes. Add float8_e4m3, float_e3m4
- Add float8_e4m3 and float_e3m4 types support - Fix float divmod with zero denominator
1 parent 6653168 commit 01d2716

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

third_party/py/ml_dtypes/ml_dtypes.BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ pybind_extension(
3838
"_src/common.h",
3939
"_src/custom_float.h",
4040
"_src/dtypes.cc",
41-
"_src/int4_numpy.h",
41+
"_src/intn_numpy.h",
4242
"_src/numpy.cc",
4343
"_src/numpy.h",
4444
"_src/ufuncs.h",

third_party/py/ml_dtypes/workspace.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ float8 varieties, and int4.
77
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
88

99
def repo():
10-
ML_DTYPES_COMMIT = "24084d9ed2c3d45bf83b7a9bff833aa185bf9172"
11-
ML_DTYPES_SHA256 = "c916a3e6b3d9bdcb476f506fdbbecb6d5e9f21f82f221dfcb42b320b4e85e55a"
10+
ML_DTYPES_COMMIT = "f802a63d5ef65e33978eece14464c8b02a7c269d"
11+
ML_DTYPES_SHA256 = "f789d68472cf02f548f0881255438708ed734f1ffcd6c64cd3e9ffe23478a9c5"
1212
tf_http_archive(
1313
name = "ml_dtypes",
1414
build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD",

third_party/tsl/third_party/py/ml_dtypes/ml_dtypes.BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ pybind_extension(
3838
"_src/common.h",
3939
"_src/custom_float.h",
4040
"_src/dtypes.cc",
41-
"_src/int4_numpy.h",
41+
"_src/intn_numpy.h",
4242
"_src/numpy.cc",
4343
"_src/numpy.h",
4444
"_src/ufuncs.h",

third_party/tsl/third_party/py/ml_dtypes/workspace.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ float8 varieties, and int4.
77
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
88

99
def repo():
10-
ML_DTYPES_COMMIT = "24084d9ed2c3d45bf83b7a9bff833aa185bf9172"
11-
ML_DTYPES_SHA256 = "c916a3e6b3d9bdcb476f506fdbbecb6d5e9f21f82f221dfcb42b320b4e85e55a"
10+
ML_DTYPES_COMMIT = "f802a63d5ef65e33978eece14464c8b02a7c269d"
11+
ML_DTYPES_SHA256 = "f789d68472cf02f548f0881255438708ed734f1ffcd6c64cd3e9ffe23478a9c5"
1212
tf_http_archive(
1313
name = "ml_dtypes",
1414
build_file = "//third_party/py/ml_dtypes:ml_dtypes.BUILD",

0 commit comments

Comments
 (0)