w

Author: nihui

tools/pnnx/tests/onnx/test_onnx_layout_ops.py

@@ -77,19 +77,6 @@ def test():
     import os
     os.system("../../src/pnnx test_onnx_layout_ops.onnx inputshape=[1,12,16,14] inputshape2=[1,12,48,66]")
 
-    with open("test_onnx_layout_ops.ncnn.param") as f:
-        lines = f.readlines()
-        if not any(line.startswith("Permute") and " out17 " in line and "0=2" in line for line in lines):
-            return False
-        if not any(line.startswith("ExpandDims") and " out10 " in line and "-23303=2,2,0" in line for line in lines):
-            return False
-        if not any(line.startswith("ExpandDims") and " out12 " in line and "-23303=2,1,2" in line for line in lines):
-            return False
-
-    with open("test_onnx_layout_ops_ncnn.py") as f:
-        if "out17.numpy(batch_index=233)" in f.read():
-            return False
-
     # pnnx inference
     import test_onnx_layout_ops_pnnx
     b = test_onnx_layout_ops_pnnx.test_inference()

tools/pnnx/tests/test_Tensor_reshape.py

@@ -58,19 +58,6 @@ def test():
     import os
     os.system("../src/pnnx test_Tensor_reshape.pt inputshape=[1,3,16],[1,5,9,11],[14,8,5,9,10],[210],[2,3,5,7],[280],[2,3,5,7],[210]")
 
-    with open("test_Tensor_reshape.ncnn.param") as f:
-        lines = f.readlines()
-        if not any(line.startswith("Reshape") and "12=1" in line for line in lines):
-            return False
-        if not any(line.startswith("Reshape") and "12=2" in line for line in lines):
-            return False
-        if not any(line.startswith("Permute") for line in lines):
-            return False
-        if not any(line.startswith("Reshape") and "0=140" in line and "12=" not in line for line in lines):
-            return False
-        if any(line.startswith("Reshape") and "12=2" in line and "=-1" in line for line in lines):
-            return False
-
     # pnnx inference
     import test_Tensor_reshape_pnnx
     b = test_Tensor_reshape_pnnx.test_inference()

tools/pnnx/tests/test_torch_flatten.py

@@ -40,17 +40,6 @@ def test():
     import os
     os.system("../src/pnnx test_torch_flatten.pt inputshape=[1,3,16],[1,5,9,11],[14,8,5,9,10],[2,3,5,7]")
 
-    with open("test_torch_flatten.ncnn.param") as f:
-        lines = f.readlines()
-        if sum(1 for line in lines if line.startswith("Reshape") and "12=1" in line) != 2:
-            return False
-        if not any(line.startswith("Reshape") and "0=-1" in line and "12=1" in line for line in lines):
-            return False
-        if not any(line.startswith("Reshape") and "0=7" in line and "1=5" in line and "2=-1" in line and "12=1" in line for line in lines):
-            return False
-        if not any(line.startswith("Reshape") and "0=35" in line and "1=4" in line and "2=" not in line for line in lines):
-            return False
-
     # pnnx inference
     import test_torch_flatten_pnnx
     b = test_torch_flatten_pnnx.test_inference()