Update for xformers (#2372)

* update api

* update kernel api

* clang-format
This commit is contained in:
Max Podkorytov
2025-06-22 00:28:30 -07:00
committed by GitHub
parent cebdee4d9e
commit 0366fb2abc
2 changed files with 18 additions and 2 deletions

View File

@@ -847,6 +847,7 @@ struct FmhaFwdKernel
window_size_left,
window_size_right,
mask_type,
0, // min_seqlen_q
p_drop,
s_randval,
std::make_pair(std::get<0>(drop_seed_offset), std::get<1>(drop_seed_offset)));