Add receipt for pytorch

alugorey · alugorey · commit b9151ca3b1a3 · 2025-08-08T19:04:10.000Z
diff --git a/csrc/cpp_itfs/mha_fwd_generate.py b/csrc/cpp_itfs/mha_fwd_generate.py
@@ -155,6 +155,8 @@
     5: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API)
     + FMHA_FWD_SPLITKV_API
     + FMHA_BATCH_PREFILL_API,
+    6: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API)
+    + FMHA_FWD_SPLITKV_API,
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -155,6 +155,8 @@`
`155`	`155`	`5: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API)`
`156`	`156`	`+ FMHA_FWD_SPLITKV_API`
`157`	`157`	`+ FMHA_BATCH_PREFILL_API,`
	`158`	`+ 6: FMHA_FWD_API.format(F_inner_dispatch=COMBINED_API)`
	`159`	`+ + FMHA_FWD_SPLITKV_API,`
`158`	`160`	`}`
`159`	`161`
`160`	`162`