[Minor] Fix styles for overlap mode (#2068)

2024-11-17 19:49:20 -08:00
parent 8c280cee55
commit a9e90b4bce
4 changed files with 8 additions and 17 deletions
--- a/test/srt/test_triton_attention_backend.py
+++ b/test/srt/test_triton_attention_backend.py
@@ -1,4 +1,8 @@
-import subprocess
+"""
+Usage:
+python3 -m unittest test_triton_attention_backend.TestTritonAttnBackend.test_mmlu
+"""
+
 import unittest
 from types import SimpleNamespace