Skip to content

Commit 7fed498

Browse files
gpires-metafacebook-github-bot
authored andcommitted
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
1 parent de56c81 commit 7fed498

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/llm/custom_ops/test_sdpa_with_kv_cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ def setUp(self):
282282
self.is_causal = True
283283

284284

285-
class SDPAWithDynamicShape(unittest.TestCase):
285+
class SDPAWithDynamicShapeTest(unittest.TestCase):
286286

287287
def setUp(self):
288288
torch.manual_seed(42)

0 commit comments

Comments
 (0)