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

..

__init__.py

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

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

cuda_wrapper.py

update flake8 version to support pre-commit in python3.12 (#3000)

2025-07-24 01:43:31 -07:00

custom_all_reduce.py

Add custom op declaration for all_reduce (#3473)

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