fix lint

pytorch-labs · Dec 9, 2024 · 5aa17b6 · 5aa17b6
1 parent cdcd06d
commit 5aa17b6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tritonbench/utils/triton_op.py b/tritonbench/utils/triton_op.py
@@ -623,6 +623,7 @@ def __init__(
         if self.mode in [Mode.FWD_BWD, Mode.BWD]:
             # TODO: remove this once we have a better way to handle backward benchmarking
             import torch._functorch.config
+
             torch._functorch.config.donated_buffer = False
         self.device = tb_args.device
         self.required_metrics = (