xla
3385bd64 - Refactor Dynamo (custom op) integration code (#5805)

Commit

1 year ago

Refactor Dynamo (custom op) integration code (#5805) * Refactor and clean SPMD+Dynamo integration code

References

#5805 - Refactor Dynamo (custom op) integration code

Author

yeounoh

yeounoh

Parents

Files5

test/spmd
- test_dynamo_spmd.py
torch_xla
- csrc
  - init_python_bindings.cpp
  - xla_sharding_util.cpp
  - xla_sharding_util.h
- distributed/spmd
  - xla_sharding.py