xla
switch to pinning layout by default in FSDP to avoid data corruption in PJRT and be consistent with xm.all_reduce
#4359
Merged
Merged
switch to pinning layout by default in FSDP to avoid data corruption in PJRT and be consistent with xm.all_reduce #4359
JackCaoG
merged
1 commit into
pytorch:master
from
ronghanghu:xla_fsdp_pin_layout_by_default
switch to pinning layout by default in FSDP to avoid data corruption …
5aded050
JackCaoG
approved these changes
on 2022-12-16
Login to write a write a comment.
Login via GitHub