tests/pytorch/test_cuda_graphs.py · 30c0120b34c9d2fbb78e454167beb122ad44ab35 · OpenDAS / TransformerEngine

"...git@developer.sourcefind.cn:jerrrrry/infinicore.git" did not exist on "3e5842c3110d0a8274fdb885394d59018b72e84c"

[PyTorch] Add reset cudagraph interface (#2367) · 262c184e

Robin Zhang authored Nov 14, 2025



* reset cudagraph
Signed-off-by: Robin Zhang <robinz@nvidia.com>

* use closure instead of mutable default values
Signed-off-by: Robin Zhang <robinz@nvidia.com>

* add test
Signed-off-by: Robin Zhang <robinz@nvidia.com>

* fix test
Signed-off-by: Robin Zhang <robinz@nvidia.com>

---------
Signed-off-by: Robin Zhang <robinz@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

262c184e

test_cuda_graphs.py 22.1 KB

Replace test_cuda_graphs.py