forked from ccf-ai-infra/TileOPs-Metax
6a61ba14d9
## Summary Align `kernels/`, `tests/`, and `benchmarks/` file structure 1:1 with the post-#890 `ops/` layout. Eliminates unnecessary directory nesting and consolidates trivially small files. **Changes (166 files, 11 commits):** 1. Consolidate attention kernel directories into `kernels/attention/` 2. Rename `kernels/conv/` to `kernels/convolution/` 3. Drop `_fwd` suffix from mamba kernel filenames 4. Flatten `kernels/norm/` subdirectories 5. Flatten `kernels/reduction/` subdirectories 6. Flatten single-file kernel directories 7. Consolidate tests and benchmarks to mirror kernel layout 8. Delete stale `CHANGELOG.md` Closes #924 ## Test plan - [x] **AC-1**: All `__init__.py`, `ops_manifest.yaml`, and internal imports updated for moved files -- Manifest path audit: checked 208 paths, 0 missing. Runtime validation passed. - [x] **AC-2**: `from tileops.kernels import ...` resolves for all moved kernels -- 73 symbols exported and resolved with 0 failures. - [x] **AC-3**: Full test suite passes -- `pytest tests` -> 2361 passed, 22 skipped, 0 failed (236.78s). - [x] **AC-4**: No orphaned directories remain after migration -- `find` produced no empty directories. ## Follow-up - #929 — Update stale kernel path reference in docs/manifest.md (doc drift from this refactor) --------- Co-authored-by: Ibuki 🍃 — a wind born from Claude Opus <Ibuki-wind@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| attention | ||
| moe | ||
| norm | ||
| reduction | ||
| __init__.py | ||
| convolution.py | ||
| da_cumsum.py | ||
| deltanet.py | ||
| deltanet_recurrence.py | ||
| dropout.py | ||
| elementwise.py | ||
| engram.py | ||
| engram_decode.py | ||
| fft.py | ||
| fp8_lighting_indexer.py | ||
| fp8_quant.py | ||
| gated_deltanet.py | ||
| gated_deltanet_recurrence.py | ||
| gemm.py | ||
| gla.py | ||
| gla_recurrence.py | ||
| grouped_gemm.py | ||
| op.py | ||
| pool.py | ||
| rope.py | ||
| ssd_chunk_scan.py | ||
| ssd_chunk_state.py | ||
| ssd_decode.py | ||
| ssd_state_passing.py | ||
| topk_selector.py | ||