xla
Support `dist.all_gather` related collective ops
#7860

Merged

Commits

add test case

zpcore committed 2 years ago
nit

zpcore committed 2 years ago
nit

zpcore committed 2 years ago
clean up test and add dist.all_gather

zpcore committed 2 years ago
remove mock object

zpcore committed 2 years ago
nit

zpcore committed 2 years ago
update to openxla backend

zpcore committed 2 years ago
use toy backend

zpcore committed 2 years ago

Loading