Fix other bwd

pytorch-labs · Nov 18, 2024 · 1f4fa82 · 1f4fa82
1 parent 90bbb73
commit 1f4fa82
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/run.py b/run.py
@@ -55,12 +55,6 @@ def _run(args: argparse.Namespace, extra_args: List[str]) -> BenchmarkOperatorRe
         Opbench = load_opbench_by_name_from_loader(args)
     else:
         Opbench = load_opbench_by_name(args.op)
-    if args.fwd_bwd:
-        args.mode = "fwd_bwd"
-    if args.bwd:
-        args.mode = "bwd"
-    if args.fwd_no_grad:
-        args.mode = "fwd_no_grad"
     opbench = Opbench(
         tb_args=args,
         extra_args=extra_args,