Skip to content

Commit f1601fb

Browse files
justinjfuGoogle-ML-Automation
authored andcommitted
Delete jax._src.dtypes.bit_width from internal API.
PiperOrigin-RevId: 836308191
1 parent 5b5f09e commit f1601fb

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

jax/_src/dtypes.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,9 +283,6 @@ def itemsize_bits(dtype: DTypeLike) -> int:
283283
else:
284284
raise ValueError(f"unexpected input: {dtype=}")
285285

286-
# TODO(justinfu): Rename all instances of bit_width to itemsize_bits
287-
bit_width = itemsize_bits
288-
289286
# Trivial vectorspace datatype needed for tangent values of int/bool primals
290287
float0: np.dtype = np.dtype([('float0', np.void, 0)])
291288

0 commit comments

Comments
 (0)