Fix lints

pytorch · Sep 12, 2024 · 48cb738 · 48cb738
1 parent 3366be2
commit 48cb738
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/torchbenchmark/operators/addmm/hstu.py b/torchbenchmark/operators/addmm/hstu.py
@@ -6,8 +6,6 @@
 import triton
 from torchbenchmark import add_path, SUBMODULE_PATH
 
-from typing import Tuple
-
 with add_path(str(SUBMODULE_PATH)):
     triton_addmm = importlib.import_module(
         "generative-recommenders.ops.triton.triton_addmm"

diff --git a/torchbenchmark/operators/ragged_attention/operator.py b/torchbenchmark/operators/ragged_attention/operator.py
@@ -34,7 +34,7 @@ def __init__(
     def hstu_triton_ragged_attention(self, qkv, seq_offsets, timestamps):
         attn = RaggedHSTUAttn(self.batch_size, self.num_heads, self.max_seq_len, self.num_buckets, persistent_kernel=False)
         return lambda: attn(qkv, seq_offsets, timestamps)
-    
+
     @register_benchmark()
     def hstu_triton_ragged_attention_persistent(self, qkv, seq_offsets, timestamps):
         attn = RaggedHSTUAttn(self.batch_size, self.num_heads, self.max_seq_len, self.num_buckets, persistent_kernel=True)