diff --git a/tests/unit/test_fmha_sm100.py b/tests/unit/test_fmha_sm100.py index 4f1b9ed6..90eb35c7 100644 --- a/tests/unit/test_fmha_sm100.py +++ b/tests/unit/test_fmha_sm100.py @@ -32,7 +32,7 @@ def compile_kernel(): f"{CUDA}/bin/nvcc", "--std=c++20", "-shared", - "-fPIC", + "-Xcompiler", "-fPIC", f"-gencode=arch=compute_100a,code=sm_100a", f"-I{REPO}", f"-I{CUTLASS}/include",