tests/pytorch/test_cuda_graphs.py · 6c57926782228fb49cbe5e52c92b35e1658ae6f5 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored Jun 13, 2024



* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Apply formatting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

9416519d

test_cuda_graphs.py 9.36 KB

Replace test_cuda_graphs.py