diff --git a/dsv4/kernels/attention/fmha_multihead_op.py b/dsv4/kernels/attention/fmha_multihead_op.py index 95f595a6..a0695f62 100644 --- a/dsv4/kernels/attention/fmha_multihead_op.py +++ b/dsv4/kernels/attention/fmha_multihead_op.py @@ -72,7 +72,7 @@ def _ensure_built(): nvcc, "-std=c++20", "-shared", - "-fPIC", + "-Xcompiler", "-fPIC", "-gencode=arch=compute_100a,code=sm_100a", "-gencode=arch=compute_100a,code=compute_100a", f"-I{KERNEL_DIR}",