refactor(//tests/py): Update QAT Testscript

narendasan · narendasan · commit fc037c2bcce5 · 2021-09-23T11:47:35.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/tests/py/test_qat_trt_accuracy.py b/tests/py/test_qat_trt_accuracy.py
@@ -55,7 +55,7 @@ def test_compile_script(self):
 
         compile_spec = {
             "inputs": [trtorch.Input([16, 3, 32, 32])],
-            "op_precision": torch.int8,
+            "enabled_precisions": {torch.int8},
             # "enabled_precision": {torch.float32, torch.int8},
         }
 

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ def test_compile_script(self):`
`55`	`55`
`56`	`56`	`compile_spec = {`
`57`	`57`	`"inputs": [trtorch.Input([16, 3, 32, 32])],`
`58`		`- "op_precision": torch.int8,`
	`58`	`+ "enabled_precisions": {torch.int8},`
`59`	`59`	`# "enabled_precision": {torch.float32, torch.int8},`
`60`	`60`	`}`
`61`	`61`