mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Deprecate jax.lib.xla_client.PaddingType.

Browse Source 
This type is unused by JAX, so there is no replacement.

(JAX does have an internal PaddingType enum in lax, but it is not present in any APIs, as best I can tell.)

PiperOrigin-RevId: 684451556
This commit is contained in:
Peter Hawkins 2024-10-10 08:21:28 -07:00 committed by jax authors
parent 70ee8e1161
commit aa3254d723
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
CHANGELOG.md
View File

@ -25,6 +25,10 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
operations. The semi-public API `jax.lib.xla_client.FftType` has been
deprecated.
* Deprecations:
* The semi-public API `jax.lib.xla_client.PaddingType` has been deprecated.
No JAX APIs consume this type, so there is no replacement.
## jax 0.4.34 (October 4, 2024)
* New Functionality

9
jax/lib/xla_client.py
View File

@ -29,7 +29,6 @@ DeviceAssignment = _xc.DeviceAssignment
Frame = _xc.Frame
HloSharding = _xc.HloSharding
OpSharding = _xc.OpSharding
PaddingType = _xc.PaddingType
PrimitiveType = _xc.PrimitiveType
Shape = _xc.Shape
Traceback = _xc.Traceback
@ -63,6 +62,13 @@ _deprecations = {
"jax.lib.xla_client.FftType is deprecated; use jax.lax.FftType.",
_FftType,
),
"PaddingType": (
(
"jax.lib.xla_client.PaddingType is deprecated; this type is unused"
" by JAX so there is no replacement."
),
_xc.PaddingType,
),
}
import typing as _typing
@ -72,6 +78,7 @@ if _typing.TYPE_CHECKING:
bfloat16 = _xc.bfloat16
Device = _xc.Device
FftType = _FftType
PaddingType = _xc.PaddingType
XlaRuntimeError = _xc.XlaRuntimeError
else:
from jax._src.deprecations import deprecation_getattr as _deprecation_getattr