Fix flash attention for ROCm #11353
| Job | Run time |
|---|---|
| 54s | |
| 44s | |
| 2m 49s | |
| 1m 22s | |
| 2m 34s | |
| 2m 49s | |
| 2m 32s | |
| 41s | |
| 1m 43s | |
| 5m 41s | |
| 5m 36s | |
| 2m 13s | |
| 6m 53s | |
| 1m 27s | |
| 1m 44s | |
| 1m 19s | |
| 4m 29s | |
| 3m 38s | |
| 6m 44s | |
| 17m 0s | |
| 15m 0s | |
| 6m 3s | |
| 3m 38s | |
| 17m 28s | |
| 6m 15s | |
| 8m 30s | |
| 19m 55s | |
| 7m 42s | |
| 4m 25s | |
| 5m 30s | |
| 6m 15s | |
| 5m 11s | |
| 6m 28s | |
| 0s | |
| 3h 5m 12s |