[TRITON] Move triton_tests into respective folders

[Boss2002n]

Reorganize triton_tests into categorized folder structure

Restructured the triton_tests directory to improve code organization and maintainability by grouping related tests into logical categories.

Easier to run specific test categories, for example:
pytest aiter/op_tests/triton_tests/ # All tests
pytest aiter/op_tests/triton_tests/gemm/ # Just GEMM tests
pytest aiter/op_tests/triton_tests/gemm/basic/ # Just basic GEMM tests
pytest aiter/op_tests/triton_tests/moe/ # Just MOE tests
and so on

[Copilot]

Pull request overview

This PR reorganizes the triton test files by moving them into a more structured folder hierarchy, updating all import statements to reflect the new paths.

Key Changes:

• Updated import statements across multiple test files to use absolute paths starting with op_tests.triton_tests
• Reorganized tests into categorized subdirectories (gemm/basic, gemm/fused, gemm/feed_forward, gemm/batched, quant, moe, rope)
• Changed relative imports to absolute imports for better clarity and maintainability

Reviewed changes

Copilot reviewed 13 out of 62 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
op_tests/triton_tests/test_fused_kv_cache.py	Updated imports to reference rope tests from new op_tests.triton_tests.rope path
op_tests/triton_tests/test_activation.py	Updated imports to reference quant and gemm tests from new nested folder structure
op_tests/triton_tests/rope/test_rope.py	Updated import to use absolute path for ref_rope functions
op_tests/triton_tests/quant/test_fused_mxfp4_quant.py	Updated imports to reference gemm and quant tests from new folder structure
op_tests/triton_tests/quant/test_fused_fp8_quant.py	Updated import to reference quant test from new path
op_tests/triton_tests/moe/test_moe_mx.py	Updated import to reference moe test from new path
op_tests/triton_tests/gemm/fused/test_fused_gemm_afp4wfp4_split_cat.py	Updated import to reference batched gemm test from new path
op_tests/triton_tests/gemm/fused/test_fused_gemm_afp4wfp4_mul_add.py	Updated imports to reference basic gemm tests from new path
op_tests/triton_tests/gemm/fused/test_fused_gemm_afp4wfp4_a16w16.py	Updated imports to reference basic gemm tests from new nested structure
op_tests/triton_tests/gemm/fused/test_fused_gemm_a8w8_blockscale_a16w16.py	Updated imports to reference basic gemm tests from new path
op_tests/triton_tests/gemm/feed_forward/test_ff_a16w16_fused.py	Updated imports to reference basic gemm and feed_forward utils from new paths
op_tests/triton_tests/gemm/feed_forward/test_ff_a16w16.py	Updated imports to reference feed_forward utils and basic gemm from new paths
op_tests/triton_tests/gemm/basic/test_gemm_a16w16_gated.py	Updated import to reference basic gemm test from new path

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

[brunomazzottiamd]

LGTM!

[lucas-santos-amd]

With a PR this big, it's a good idea to wait for the Triton CI to pass, even though it isn't required to merge. That aside, LGTM!

[brunomazzottiamd]

With a PR this big, it's a good idea to wait for the Triton CI to pass, even though it isn't required to merge. That aside, LGTM!

It's really mandatory all Triton tests to pass.

[vgokhale]

LGTM!