diff --git a/dsv4/kernels/attention/fmha_common.cuh b/dsv4/kernels/attention/fmha_common.cuh index de5e81c5..559125f0 100644 --- a/dsv4/kernels/attention/fmha_common.cuh +++ b/dsv4/kernels/attention/fmha_common.cuh @@ -49,6 +49,7 @@ #include #include +#include #include namespace dsv4::kernels::attention {