xla
add `auto_wrap_policy` into XLA FSDP for automatic wrapping
#4318
Merged

Loading