forked from ccf-ai-infra/TileOPs-Metax
4c1ddeb827
## Summary Fixes #1607. This PR restores `tests/ops -m smoke` on the new image-baked TileLang/tvm-ffi stack by addressing the import, binder, lowering, and timeout issues uncovered by the GPU smoke run. Changes include: - remove remaining `tvm.tir` smoke gates/import assumptions in GQA FP8 and topk selector paths - handle new TileLang binder argument errors in binary autotune fallback - avoid Conv1d local scalar binding that new TileLang rejects - inline GatedDeltaNet recurrence scalars that were treated as missing API args - materialize side-effecting MoE extern atomic results before indexed stores so lowering does not duplicate atomic calls - fix FP8LightingIndexer `block_Q > 1` GEMM output shape and use batched queries by default to avoid smoke timeout - make `gpu-smoke` use a run-local `TRITON_CACHE_DIR` under `/ci-cache/triton/gpu-smoke-${GITHUB_RUN_ID}-${GITHUB_RUN_ATTEMPT}` so TorchInductor/Triton writes do not collide with stale shared-cache hash directories owned by another runner user ## Validation Image: `ghcr.io/tile-ai/tileops-runner:65dbc98-torch2.10` GPU: local GPU1 (H200 visible as device 0 in container) ```text python -m pytest tests/ops/attention/test_gqa_fp8.py \ tests/ops/test_topk_selector.py \ tests/ops/test_binary_arith.py \ tests/ops/test_convolution.py \ tests/ops/test_gated_deltanet_fwd.py \ tests/ops/test_gated_deltanet_chunkwise_bwd.py \ tests/ops/test_moe_permute_nopad.py \ tests/ops/test_moe_permute_align.py \ tests/ops/test_moe_fused_moe.py \ tests/ops/test_fp8_lighting_indexer.py \ -m smoke -q --tb=short --disable-warnings 162 passed, 102 deselected python -m pytest tests/ops -m smoke -q --tb=short --disable-warnings 2180 passed, 24 skipped, 955 deselected, 283 warnings in 148.30s TRITON_CACHE_DIR=/data7/shared/ci-cache/triton/gpu-smoke-local-codex \ python -m pytest tests/ops/test_mamba.py tests/ops/test_moe_fused_moe.py \ -m smoke -q --tb=short --disable-warnings 42 passed, 21 deselected ``` |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| gqa_fp8_utils.py | ||