Skip to content

Commit a3a74e2

Browse files
deqiangcGoogle-ML-Automation
authored andcommitted
Link HloSharding ser-deser support to common_serdes
PiperOrigin-RevId: 708364664
1 parent edcdfcf commit a3a74e2

File tree

1 file changed

+1
-0
lines changed
  • xla/python/ifrt_proxy/common

1 file changed

+1
-0
lines changed

xla/python/ifrt_proxy/common/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,7 @@ cc_library(
173173
"//xla/python/ifrt:plugin_program_serdes",
174174
"//xla/python/ifrt/hlo:hlo_program_serdes",
175175
"//xla/python/ifrt/ir:ifrt_ir_program_serdes",
176+
"//xla/python/pjrt_ifrt:xla_sharding_serdes",
176177
],
177178
alwayslink = True,
178179
)

0 commit comments

Comments
 (0)