Move lhs_axis and rhs_axis

sfvaroglu · sfvaroglu · commit 5bd637d74d4b · 2022-04-01T16:48:08.000-07:00
diff --git a/tests/python/contrib/test_arm_compute_lib/test_add.py b/tests/python/contrib/test_arm_compute_lib/test_add.py
@@ -38,8 +38,6 @@
     "rhs_zero_point": relay.const(85, "int32"),
     "output_scale": relay.const(0.0235294, "float32"),
     "output_zero_point": relay.const(128, "int32"),
-    "lhs_axis": int(-1),
-    "rhs_axis": int(-1),
 }
 
 
@@ -76,6 +74,10 @@ def _get_expected_codegen(shape, dtype, op_name, qnn_params):
         },
     }
 
+    if qnn_params:
+        node["lhs_axis"] = [[-1]]
+        node["rhs_axis"] = [[-1]]
+
     return [*inputs, node]
 
 

Original file line number	Diff line number	Diff line change
`@@ -38,8 +38,6 @@`
`38`	`38`	`"rhs_zero_point": relay.const(85, "int32"),`
`39`	`39`	`"output_scale": relay.const(0.0235294, "float32"),`
`40`	`40`	`"output_zero_point": relay.const(128, "int32"),`
`41`		`- "lhs_axis": int(-1),`
`42`		`- "rhs_axis": int(-1),`
`43`	`41`	`}`
`44`	`42`
`45`	`43`
`@@ -76,6 +74,10 @@ def _get_expected_codegen(shape, dtype, op_name, qnn_params):`
`76`	`74`	`},`
`77`	`75`	`}`
`78`	`76`
	`77`	`+ if qnn_params:`
	`78`	`+ node["lhs_axis"] = [[-1]]`
	`79`	`+ node["rhs_axis"] = [[-1]]`
	`80`	`+`
`79`	`81`	`return [*inputs, node]`
`80`	`82`
`81`	`83`