[Test] Consolidate kernel tests under plural kernels tree (#39966)
This commit is contained in:
+1
-1
@@ -191,7 +191,7 @@ are comparable.
|
||||
| `../subblock_sparse_attn.py` | the `AttentionBackend`: schedule, gating, dense fallback |
|
||||
|
||||
Tests: `test/unit/test_subblock_sparse_attention.py` and
|
||||
`test/registered/kernel/attention/test_subblock_sage_fp8_sm90.py`. The GPU
|
||||
`test/registered/kernels/ops/attention/test_subblock_sage_fp8_sm90.py`. The GPU
|
||||
test covers the native production dispatch. Running at a full block budget must
|
||||
reproduce dense attention up to the expected quantization error, pinning routing
|
||||
indices, ragged tails, scale domains and the softmax scale in one check.
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
"""Hand-tuned dispatch configs for the JIT custom all-reduce (v2).
|
||||
|
||||
Thresholds and block counts come from sweeps of
|
||||
``test/registered/jit/benchmark/bench_custom_all_reduce.py`` on the listed
|
||||
GPUs; ``get_all_reduce_config`` picks the table for the current arch and
|
||||
world size.
|
||||
``test/registered/kernels/benchmark/communication/bench_custom_all_reduce.py``
|
||||
on the listed GPUs; ``get_all_reduce_config`` picks the table for the current
|
||||
arch and world size.
|
||||
"""
|
||||
|
||||
from functools import cache
|
||||
|
||||
Reference in New Issue
Block a user