Support skip-softmax attention (#19089)

This commit is contained in:
Shu Wang
2026-03-28 15:55:48 -07:00
committed by GitHub
parent 40ff652862
commit efebcab43e
8 changed files with 122 additions and 0 deletions
+1
View File
@@ -1937,6 +1937,7 @@ if __name__ == "__main__":
"mmmu",
"image",
"mooncake",
"longbench_v2",
],
help="Name of the dataset to benchmark on.",
)