Skip to content

Commit c329e0a

Browse files
gneculaGoogle-ML-Automation
authored andcommitted
[export] Remove serialization of old 16-bit nr_devices
We added a 32-bit version of nr_devices on 11/23/2025 but I kept serializating the 16-bit version until now, in case there are old deserializers still running. The forward compatibility window of 30-days has passed and we can stop serializing nr_devices_short. PiperOrigin-RevId: 861832454
1 parent 191504b commit c329e0a

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

jax/_src/export/serialization.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,6 @@ def _serialize_exported(
144144
ser_flatbuf.ExportedAddOutTree(builder, out_tree)
145145
ser_flatbuf.ExportedAddOutAvals(builder, out_avals)
146146
ser_flatbuf.ExportedAddNrDevices(builder, exp.nr_devices)
147-
ser_flatbuf.ExportedAddNrDevicesShort(builder, exp.nr_devices) # For forward compatibility, can remove after January 2026
148147
ser_flatbuf.ExportedAddInShardings(builder, in_shardings)
149148
ser_flatbuf.ExportedAddOutShardings(builder, out_shardings)
150149
ser_flatbuf.ExportedAddPlatforms(builder, platforms)

0 commit comments

Comments
 (0)