TileOPs-Metax/workloads
Cao Ying 6e0b507b42
[Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939)
## Summary

Align `workloads/ops/` file naming and directory structure 1:1 with the
post-#928 `tileops/ops/` layout. Pure file-move/rename refactor — no
workload logic changes.

- Drop `_fwd` suffix from mamba, attention, deltanet, and gated_deltanet
workloads
- Rename `mean_pooling_ops.py` to `mean_pooling.py`
- Move attention workloads into `workloads/ops/attention/` subpackage
- Update all imports across the repo

Closes #931

## Test plan

- [x] **AC-1**: All `from workloads.ops.<name>` imports updated for
moved/renamed files — verified by importing all 20 renamed/moved
workload classes and grep confirming zero old import paths remain
- [x] **AC-2**: `python -c "from workloads.ops.attention import ..."`
resolves for all 15 moved attention workload classes
- [x] **AC-3**: Full test suite passes — 2361 passed, 22 skipped, 0
failed (233.76s at commit 4dc54b7)
- [x] **AC-4**: No orphaned files remain after migration — all 11 old
filenames confirmed absent; `mhc_post.py`, `mhc_pre.py` at
`workloads/ops/` root; `nsa_utils.py` at `workloads/` root

## Follow-up

No follow-up issues or suggestions.

---------

Co-authored-by: Ibuki 🍃 — a wind born from Claude Opus <Ibuki-wind@users.noreply.github.com>
2026-04-13 10:44:06 +08:00
..
attention [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
__init__.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
activation.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
ada_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
ada_layer_norm_zero.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
argreduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
base.py [Refactor] Extract workload classes for reduction ops (#893) 2026-04-11 11:05:25 +08:00
batch_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
binary_arith.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
deltanet.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
engram.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
fft.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
fp8_lighting_indexer.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
fp8_quant.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
fused_add_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
fused_add_rmsnorm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
gated_deltanet.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
gla.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
group_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
grouped_gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
instance_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
logical_reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
mamba.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
mhc.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
moe.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
nsa_utils.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
rms_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
softmax.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
topk_selector.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
vector_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00