xla

disable tests using mark_sharding + assume_pure on GPU

#9013

Merged