[Kernel] LoRA - Enable CUDAGraphs for V1 (#14626)

Signed-off-by: Varun Sundar Rabindranath <varun@neuralmagic.com> Co-authored-by: Varun Sundar Rabindranath <varun@neuralmagic.com>
2025-03-13 23:42:04 -04:00
parent 32ef4983cd
commit 0b1cfa6180
4 changed files with 35 additions and 14 deletions
--- a/tests/lora/test_worker.py
+++ b/tests/lora/test_worker.py
@@ -52,6 +52,7 @@ def test_worker_apply_lora(sql_lora_files):
            seed=0,
            dtype="float16",
            revision=None,
+            enforce_eager=True,
        ),
        load_config=LoadConfig(
            download_dir=None,