From 7fed498a7612bf1a2caa374ea7d3d887085f5716 Mon Sep 17 00:00:00 2001 From: Gabriel Pires da Silva Date: Fri, 24 Oct 2025 04:25:17 -0700 Subject: [PATCH] llm/custom_ops: follow test code conventions Summary: Rename Test Suites in `test_sdpa_with_kv_cache` to follow internal tooling conventions and make test target static listable. Previously, some of the Test Suites in the modified test file did not satisfy the condition: * Test Suite's name must contain `Test` This diff makes it so that the Test Suites follow the expected conventions. ### Benefits Following the conventions grants the test targets: * Faster test listing * Better dev experience in VSCode Differential Revision: D85428922 --- extension/llm/custom_ops/test_sdpa_with_kv_cache.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extension/llm/custom_ops/test_sdpa_with_kv_cache.py b/extension/llm/custom_ops/test_sdpa_with_kv_cache.py index 310c5b64bdf..502a6238a7d 100644 --- a/extension/llm/custom_ops/test_sdpa_with_kv_cache.py +++ b/extension/llm/custom_ops/test_sdpa_with_kv_cache.py @@ -282,7 +282,7 @@ def setUp(self): self.is_causal = True -class SDPAWithDynamicShape(unittest.TestCase): +class SDPAWithDynamicShapeTest(unittest.TestCase): def setUp(self): torch.manual_seed(42)