fixing issue 71

Summary:
TSIA

#71

Reviewed By: xuzhao9

Differential Revision: D66368132

fbshipit-source-id: fd6db21c601270ab5c1380861f64452ab419befb

tritonbench/operator_loader/__init__.py

@@ -5,7 +5,6 @@
 
 import torch
 from torch._dynamo.backends.cudagraphs import cudagraphs_inner
-from torch._inductor.compile_fx import compile_fx
 from torch._inductor.utils import gen_gm_and_inputs
 from torch._ops import OpOverload
 from torch.utils._pytree import tree_map_only
@@ -102,6 +101,9 @@ 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
                     )