Ryan bcdfc1d6b9 Add custom op declaration for all_reduce (#3473) ...

* add custom op declaration

* roll back try except

2025-08-20 20:29:58 +08:00

..

custom_all_reduce

Add custom op declaration for all_reduce (#3473)

2025-08-20 20:29:58 +08:00

__init__.py

polish code with new pre-commit rule (#2923)

2025-07-19 23:19:27 +08:00

communication.py

delete parallel_state.py (#3250)

2025-08-08 11:03:29 +08:00