Fix LTX2 RoPE JIT kernel CI (#30278)

Co-authored-by: Brayden Zhong <brayden@radixark.ai>
2026-07-07 10:15:21 +08:00
parent 6279805962
commit 267ff1b5f9
2 changed files with 9 additions and 14 deletions
@@ -3,9 +3,8 @@
 // Developed with MIT HAN Lab Kernel Design Agents:
 // https://github.com/mit-han-lab/kernel-design-agents
 //
-// This mirrors the LTX2 eager oracle:
-//   torch.nn.RMSNorm(input) returns fp32 under bf16 autocast, then split RoPE
-//   runs in fp32 and rounds once to bf16 at the final attention input.
+// This mirrors the LTX2 eager oracle: RMSNorm and split RoPE both run in
+// fp32, rounding to bf16 only once at the final attention input.

 #pragma once