remove unused args

pytorch · Oct 8, 2024 · d6bfcd1 · d6bfcd1
1 parent cc7936c
commit d6bfcd1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/torchbenchmark/operator_loader/__init__.py b/torchbenchmark/operator_loader/__init__.py
@@ -51,7 +51,7 @@ def load_opbench_by_name_from_loader(args: argparse.Namespace):
         raise ValueError(f"{args.op} is not found in the operator loader.")
     # args.op is a string, we need to evaluate it to get the actual operator overload
     op_eval = eval(args.op)
-    return dynamically_create_aten_op_class(op_eval, args)
+    return dynamically_create_aten_op_class(op_eval)
 
 
 def create_operator_class(op_eval: OpOverload):
@@ -125,9 +125,7 @@ def inductor(self, input):
     return new_class
 
 
-def dynamically_create_aten_op_class(
-    op_eval: OpOverload, args: argparse.Namespace
-):
+def dynamically_create_aten_op_class(op_eval: OpOverload):
     """
     To keep same with custom operators, we dynamically create aten operator classes here.
     """