diff --git a/tests/unit/test_fmha_sm100.py b/tests/unit/test_fmha_sm100.py index 7954b979..3ad241f5 100644 --- a/tests/unit/test_fmha_sm100.py +++ b/tests/unit/test_fmha_sm100.py @@ -29,7 +29,7 @@ print("Step 1: nvcc syntax check") print("=" * 60) nvcc_cmd = [ - "nvcc", + "/usr/local/cuda-13.2/bin/nvcc", "--std=c++17", "-gencode=arch=compute_100a,code=sm_100a", f"-I{CUTLASS}/include",