Formatting issue

pytorch · Sep 12, 2024 · 248bc3a · 248bc3a
1 parent 5c33dec
commit 248bc3a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/torchbenchmark/operators/addmm/hstu.py b/torchbenchmark/operators/addmm/hstu.py
@@ -9,9 +9,12 @@
 from typing import Tuple
 
 with add_path(str(SUBMODULE_PATH)):
-    triton_addmm = importlib.import_module("generative-recommenders.ops.triton.triton_addmm")
+    triton_addmm = importlib.import_module(
+            "generative-recommenders.ops.triton.triton_addmm"
+    )
     _addmm_fwd = triton_addmm._addmm_fwd
 
+
 class _AddMmFunction(torch.autograd.Function):
     @staticmethod
     # pyre-ignore[14]
@@ -77,6 +80,7 @@ def backward(
 
         return dx, dw, dy
 
+
 @torch.fx.wrap
 def triton_addmm(
     input: torch.Tensor,

diff --git a/torchbenchmark/operators/ragged_attention/__init__.py b/torchbenchmark/operators/ragged_attention/__init__.py
@@ -1 +1 @@
-from .operator import Operator
+from .operator import Operator
diff --git a/torchbenchmark/operators/ragged_attention/operator.py b/torchbenchmark/operators/ragged_attention/operator.py
@@ -46,4 +46,4 @@ def get_x_val(self, example_inputs):
     def get_input_iter(self):
         for _input_id in range(self._num_inputs):
             inputs = get_test_inputs(self.batch_size, self.num_heads, self.max_seq_len)
-            yield inputs
+            yield inputs