xla

aec27305 - Let grad ckpt apply opt-barrier to all params and buffers (#7206)

Let grad ckpt apply opt-barrier to all params and buffers (#7206) Summary: Let grad ckpt apply opt-barrier to all params and buffers. Test Plan: python test/test_operations.py -v -k test_opt_barrier