Small changes to improve blackwell_fmha_test.py (#4896) #6019
| Job | Run time |
|---|---|
| 6s | |
| 4s | |
| 1h 56m 4s | |
| 59m 9s | |
| 1h 53m 59s | |
| 1h 1m 48s | |
| 56m 39s | |
| 57m 23s | |
| 1h 59m 23s | |
| 55m 14s | |
| 1h 59m 1s | |
| 57m 41s | |
| 57m 45s | |
| 1h 58m 2s | |
| 1h 56m 58s | |
| 1h 58m 12s | |
| 25s | |
| 21s | |
| 26s | |
| 24s | |
| 19s | |
| 20s | |
| 24s | |
| 18s | |
| 26s | |
| 22s | |
| 22s | |
| 16s | |
| 17s | |
| 16s | |
| 20h 32m 24s |