Commit 113ef13
committed
[misc] QAT finetune: preflight-probe the fake-quant train kernel
is_attn_qat_train_available() swallows the ImportError and the selector then
filters ATTN_QAT_TRAIN out silently, so attention falls back to Flash (non-QAT)
BEFORE the cuda.py hard-fail guard fires. Add a preflight that imports
fastvideo_kernel.triton_kernels.attn_qat_train with the real traceback and
aborts if it's missing — no more silent non-QAT runs.1 parent 175bc4b commit 113ef13
1 file changed
Lines changed: 18 additions & 0 deletions
Lines changed: 18 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
42 | 42 | | |
43 | 43 | | |
44 | 44 | | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
45 | 63 | | |
46 | 64 | | |
47 | 65 | | |
| |||
0 commit comments