[FX] Fix type of argument min_acc_module_size

cgufb · facebook-github-bot · commit 750174a53adf · 2022-03-28T21:24:27.000-07:00
Summary:
As title, otherwise the below error is thrown:
```
TypeError: '&gt;=' not supported between instances of 'int' and 'str'
```

Test Plan: easy

Differential Revision: D35206473

fbshipit-source-id: f88bb0b0c314c3228d7dc3bbb7b2626692a77f58
diff --git a/torch/fx/passes/splitter_base.py b/torch/fx/passes/splitter_base.py
@@ -35,6 +35,7 @@ def __init__(self):
         parser.add_argument(
             "--min_acc_module_size",
             default=1,
+            type=int,
             help="Minimum size limit of an accelerator subgraph.",
         )
         parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,7 @@ def __init__(self):`
`35`	`35`	`parser.add_argument(`
`36`	`36`	`"--min_acc_module_size",`
`37`	`37`	`default=1,`
	`38`	`+ type=int,`
`38`	`39`	`help="Minimum size limit of an accelerator subgraph.",`
`39`	`40`	`)`
`40`	`41`	`parser.add_argument(`