[CI] Build aarch64 kernels for sgl-kernel test (#12480)

This commit is contained in:
Baizhou Zhang
2025-11-01 11:55:42 -07:00
committed by GitHub
parent 69193f7122
commit 566ade0388
3 changed files with 49 additions and 6 deletions
+2
View File
@@ -105,3 +105,5 @@ m.impl("fwd", torch::kCUDA, make_pytorch_shim(&mha_fwd));
## FAQ
- Q: Segmentation fault with CUDA 12.6
- A: Update ptxas to 12.8, reference: [segment fault error](https://github.com/Dao-AILab/flash-attention/issues/1453)
Trigger arm build test (Should be removed later)