Commit Graph
5 Commits
Author SHA1 Message Date
SolitaryThinker 2e4878c5e6 [infra]: leave FlashAttention out of the GB10 image
FA2 gives no measured speedup on GB10, and with it installed FastH3 V2
on a Spark produces different audio and frames than a native Spark
install (which has no FA2). Without it the image matches the native
install bit for bit at the same speed. Add INSTALL_FLASH_ATTN (default
1), set it to 0 for the CUDA 13 arm64 leg, and document the first-run
FlashInfer compile (persist the cache, cap MAX_JOBS).
2026-10-19 11:15:23 +00:00
SolitaryThinker fba50356b1 [infra]: skip the FA4 cute overlay in the GB10 image
FA4 cute supports SM 9/10/11 only, and the pinned revision forces
nvidia-cutlass-dsl==4.6.0.dev0 over the 4.8.x that flashinfer 0.7.2
needs, so import flashinfer fails in the image. GB10 FP4 attention runs
on fastvideo-kernel. Add INSTALL_FA4_CUTE (default 1), set it to 0 for
the CUDA 13 arm64 leg, still remove the wheel's stale cute files, and
check that flashinfer imports when the overlay is skipped.
2026-10-19 08:13:04 +00:00
SolitaryThinker 431231a285 [infra]: build the GB10 image kernel for sm_121a
The CUDA 13 arm64 image targets DGX Spark but compiled fastvideo-kernel
for 12.1, which drops the arch-specific attn_qat_infer FP4 kernels
(FASTVIDEO_H3_VSA_FP4). Build for 12.1a, as a native Spark build does,
and update the Dockerfile and Docker guide examples.
2026-10-19 03:50:29 +00:00
William Lin b36d0ef085 [infra] Add DGX Spark and multi-architecture CUDA support (#1447) 2026-07-03 13:41:43 -07:00
Mihir Jagtap 8f5712629f [docs] port to mkdocs (#855) 2025-11-04 14:31:56 -08:00