Baorun (Lauren) Mu f85e479e66 [Feature] ViT Full CUDA Graph (#35963) ...

Signed-off-by: Baorun Mu <bmu@nvidia.com>

2026-03-23 13:01:10 +08:00

..

__init__.py

[Core] Allow full cudagraph with separate attention routines and orthogonal to compilation, add support for FA2 and FlashInfer (#20059)

2025-08-15 10:01:39 -04:00

test_cudagraph_dispatch.py

[DP] Only use DP padding when cudagraphs are actually used (#34102)

2026-02-27 15:14:31 -05:00

test_cudagraph_mode.py

[Attention] Update tests to remove deprecated env vars (#30563)

2025-12-17 09:49:59 -08:00

test_encoder_cudagraph.py

[Feature] ViT Full CUDA Graph (#35963)

2026-03-23 13:01:10 +08:00