xla
Improve `assume_pure` SPMD functionality
#9360
Merged

Loading