Update tests/L0/run_fused_layer_norm/test_fused_layer_norm.py

borisfom · crcrpar · web-flow · commit 0d47550b24c9 · 2023-08-28T20:36:21.000-07:00
Co-authored-by: Masaki Kozuki &lt;mkozuki@nvidia.com&gt;
diff --git a/tests/L0/run_fused_layer_norm/test_fused_layer_norm.py b/tests/L0/run_fused_layer_norm/test_fused_layer_norm.py
@@ -270,7 +270,6 @@ def _verify_export(self, fused, fused_x):
                                                       input_names=['x_in'],
                                                       opset_version=18,
         )
-        print(onnx_str)
         assert 'x_in' in onnx_str
         assert 'ReduceMean' in onnx_str or 'LayerNormalization' in onnx_str
 

Original file line number	Diff line number	Diff line change
`@@ -270,7 +270,6 @@ def _verify_export(self, fused, fused_x):`
`270`	`270`	`input_names=['x_in'],`
`271`	`271`	`opset_version=18,`
`272`	`272`	`)`
`273`		`- print(onnx_str)`
`274`	`273`	`assert 'x_in' in onnx_str`
`275`	`274`	`assert 'ReduceMean' in onnx_str or 'LayerNormalization' in onnx_str`
`276`	`275`