InfiniTensor/pyinfinitensor
xiaonans 1c08ba200c
[feature] add cudagraph support (#215)
* [feature] add cudagraph support

* modify code to pass the cuda_all_reduce test
2024-02-21 14:00:25 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor [feature] add cudagraph support (#215) 2024-02-21 14:00:25 +08:00
tests fix mlu some kernel registration & gather op (#210) 2024-02-01 15:02:02 +08:00
pyproject.toml Change onnx-simplifier to onnxsim to resolve build issue on xpu (#164) 2023-10-21 02:58:32 +08:00