diff --git a/tests/test_kernels/triton/test_token_attn_1.py b/tests/test_infer_ops/triton/test_token_attn_1.py
similarity index 100%
rename from tests/test_kernels/triton/test_token_attn_1.py
rename to tests/test_infer_ops/triton/test_token_attn_1.py
diff --git a/tests/test_kernels/triton/test_token_attn_2.py b/tests/test_infer_ops/triton/test_token_attn_2.py
similarity index 100%
rename from tests/test_kernels/triton/test_token_attn_2.py
rename to tests/test_infer_ops/triton/test_token_attn_2.py
diff --git a/tests/test_kernels/triton/test_token_attn_fwd.py b/tests/test_infer_ops/triton/test_token_attn_fwd.py
similarity index 100%
rename from tests/test_kernels/triton/test_token_attn_fwd.py
rename to tests/test_infer_ops/triton/test_token_attn_fwd.py
diff --git a/tests/test_kernels/triton/test_token_softmax.py b/tests/test_infer_ops/triton/test_token_softmax.py
similarity index 100%
rename from tests/test_kernels/triton/test_token_softmax.py
rename to tests/test_infer_ops/triton/test_token_softmax.py