Skip to content

Add FlashAttention-3 validation helper for SM120 - #4

Merged
ivandobskygithub merged 1 commit into
mainfrom
codex/create-python-script-for-fa3-validation
Nov 20, 2025
Merged

ivandobskygithub merged 1 commit into
mainfrom
codex/create-python-script-for-fa3-validation

Conversation

@ivandobskygithub

Copy link
Copy Markdown
Owner

Summary

  • add a staged validation script under hopper/ for FlashAttention-3 SM120 setups
  • validate torch 2.9.0+cu128, CUDA 12.x capability, and the presence of FlashAttention-3 operators while rejecting FlashAttention-2
  • include a lightweight CUDA smoke test to exercise the FlashAttention-3 forward/backward path

Testing

  • Not run (not requested)

Codex Task

@ivandobskygithub
ivandobskygithub merged commit d96f44b into main Nov 20, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant