FrRay
/
TileOPs-Metax
forked from ccf-ai-infra/TileOPs-Metax
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
TileOPs-Metax/benchmarks/ops
History
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] Separate benchmark profiling from unit tests (#224) 2026-02-27 16:46:35 +08:00
bench_activation.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_ada_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_argreduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_batch_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_binary_arith.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_binary_elementwise.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_binary_strategy.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_convolution.py [Refactor] Align kernels/tests/benchmarks structure with ops layout (#928) 2026-04-12 23:24:03 +08:00
bench_cumulative.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
bench_deltanet.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_deltanet_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_dropout.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_elementwise_fp8.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_engram.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_fft.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_fp8_lighting_indexer.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_fp8_quant.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_fused_add_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_fused_add_rmsnorm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_gated_deltanet.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_gated_deltanet_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_gla_chunkwise.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_gla_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_group_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_grouped_gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_grouped_gemm_block_m.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
bench_independent_elementwise.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_instance_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_logical_reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_mamba.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_mhc_post.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_mhc_pre.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_moe_fused_moe.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_moe_fused_topk.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_moe_kernel_breakdown.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
bench_moe_permute.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_moe_permute_align.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_moe_permute_nopad.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
bench_moe_shared_fused_moe.py [Feat][MoE] SharedFusedMoE with native shared expert kernel and TP support (#778) 2026-04-07 17:42:09 +08:00
bench_moe_unpermute.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_pool.py [Refactor] Align kernels/tests/benchmarks structure with ops layout (#928) 2026-04-12 23:24:03 +08:00
bench_reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_reduce_multidim.py [Bench][Argreduce] Add dim=0 benchmark cases for 3D tensors (#913) 2026-04-12 00:52:22 +08:00
bench_rms_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_rope.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_softmax.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_topk_selector.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
bench_unary_elementwise.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_unary_strategy.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
bench_vector_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00