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/tests/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
gla_test_utils.py [Feat][Linear-Attn] Add GLA forward and backward operators (#534) 2026-03-16 21:07:55 +08:00
test_activation.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_ada_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_ada_layer_norm_zero.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_argreduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_batch_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_binary_arith.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_bitwise.py [Feat][Elementwise] Add BinaryKernel + FusedGatedKernel template ops (22 ops) (#463) 2026-03-13 13:21:27 +08:00
test_comparison.py [Feat][Elementwise] Add BinaryKernel + FusedGatedKernel template ops (22 ops) (#463) 2026-03-13 13:21:27 +08:00
test_convolution.py [Refactor] Align kernels/tests/benchmarks structure with ops layout (#928) 2026-04-12 23:24:03 +08:00
test_cumulative.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
test_deltanet_chunkwise_bwd.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_deltanet_fwd.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_deltanet_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_dropout.py [Feat][Elementwise] Add DropoutKernel + DropoutOp with Philox PRNG (#491) 2026-03-16 11:27:48 +08:00
test_engram.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fft.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fp8_lighting_indexer.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fp8_quant.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fused_add_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fused_add_rmsnorm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_fused_gated.py [Feat][Elementwise] Add strategy selection to FusedGatedKernel (#490) 2026-03-16 12:34:20 +08:00
test_gated_deltanet_chunkwise_bwd.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_gated_deltanet_fwd.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_gated_deltanet_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_gla_chunkwise_bwd.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_gla_chunkwise_fwd.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_gla_recurrence.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_group_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_grouped_gemm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_instance_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_layer_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_logical.py [Feat][Elementwise] Add BinaryKernel + FusedGatedKernel template ops (22 ops) (#463) 2026-03-13 13:21:27 +08:00
test_logical_reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_mamba.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_mhc_post.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_mhc_pre.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_moe_fused_moe.py [Refactor][MoE] Unified interface and grouped GEMM performance optimization (#706) 2026-03-31 12:49:29 +08:00
test_moe_fused_moe_distributed.py [Refactor] Align kernels/tests/benchmarks structure with ops layout (#928) 2026-04-12 23:24:03 +08:00
test_moe_fused_topk.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_moe_permute.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_moe_permute_align.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_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
test_moe_shared_fused_moe_distributed.py [Feat][MoE] SharedFusedMoE with native shared expert kernel and TP support (#778) 2026-04-07 17:42:09 +08:00
test_moe_unpermute.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_norm_ops.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_pool.py [Refactor] Align kernels/tests/benchmarks structure with ops layout (#928) 2026-04-12 23:24:03 +08:00
test_reduce.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_reduce_dim_none.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
test_reduce_multidim.py [Refactor][Naming] Rename Op/Kernel classes to match manifest keys (#873) 2026-04-09 19:57:59 +08:00
test_reduction_primitives.py [Refactor][Benchmark] Detach benchmarks from tests via workloads layer (#787) 2026-04-05 11:38:51 +08:00
test_rms_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_rope.py [CI] Speed up nightly packaging job from 31min to ~1min (#700) 2026-03-29 13:11:16 +08:00
test_softmax.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_special_elementwise.py [Fix][Elementwise] Add missing input and scalar validation to independent ops (#610) 2026-03-23 16:20:39 +08:00
test_topk_selector.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_unary_math.py [Chore][Kernels] Replace assert-based validation with ValueError and remove issue refs (#507) 2026-03-15 18:59:23 +08:00
test_vector_norm.py [Refactor][Workloads] Align workloads/ops/ naming with tileops/ops/ layout (#939) 2026-04-13 10:44:06 +08:00
test_welford_non_aligned.py [Test][Reduction] Welford non-aligned N correctness tests (#895) 2026-04-11 21:27:35 +08:00