forked from jiuyuan/InfiniTensor
1c08ba200c
* [feature] add cudagraph support * modify code to pass the cuda_all_reduce test |
||
---|---|---|
.. | ||
docs | ||
src/pyinfinitensor | ||
tests | ||
pyproject.toml |