Fix commit

pytorch-labs · Nov 26, 2024 · c061bbf · c061bbf
1 parent d542302
commit c061bbf
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tritonbench/operator_loader/__init__.py b/tritonbench/operator_loader/__init__.py
@@ -103,8 +103,6 @@ def get_input_iter(self) -> Generator:
                     "aten::convolution_backward",
                 )
                 if self.device == "cuda":
-                    from torch._inductor.compile_fx import compile_fx
-
                     cudagraph_eager = cudagraphs_inner(
                         gm, gm_args, copy_outputs=False, copy_inputs=False
                     )