Update _compiler.py

narendasan · narendasan · commit ee07cff7ddfc · 2021-11-04T08:47:02.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
diff --git a/py/torch_tensorrt/ts/_compiler.py b/py/torch_tensorrt/ts/_compiler.py
@@ -112,7 +112,7 @@ def compile(module: torch.jit.ScriptModule,
             "enabled": not require_full_compilation,
             "forced_fallback_ops": torch_executed_ops,
             "forced_fallback_modules": torch_executed_modules,
-            "min_block_size": min_block_sizexs
+            "min_block_size": min_block_size
         }
     }
 
@@ -248,4 +248,4 @@ def check_method_op_support(module: torch.jit.ScriptModule, method_name: str) ->
     Returns:
         bool: True if supported Method
     """
-    return _C.check_method_op_support(module._c, method_name)
+    return _C.check_method_op_support(module._c, method_name)

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def compile(module: torch.jit.ScriptModule,`
`112`	`112`	`"enabled": not require_full_compilation,`
`113`	`113`	`"forced_fallback_ops": torch_executed_ops,`
`114`	`114`	`"forced_fallback_modules": torch_executed_modules,`
`115`		`- "min_block_size": min_block_sizexs`
	`115`	`+ "min_block_size": min_block_size`
`116`	`116`	`}`
`117`	`117`	`}`
`118`	`118`
`@@ -248,4 +248,4 @@ def check_method_op_support(module: torch.jit.ScriptModule, method_name: str) ->`
`248`	`248`	`Returns:`
`249`	`249`	`bool: True if supported Method`
`250`	`250`	`"""`
`251`		`- return _C.check_method_op_support(module._c, method_name)`
	`251`	`+ return _C.check_method_op_support(module._c, method_name)`