From 7ec2d03465ebbb8aaa766acedc7710238a30eab7 Mon Sep 17 00:00:00 2001 From: nihuini Date: Tue, 5 Aug 2025 19:59:14 +0800 Subject: [PATCH] flip --- tools/pnnx/src/pass_level2/torch_flip.cpp | 7 ------- tools/pnnx/tests/onnx/test_torch_flip.py | 4 ++++ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/tools/pnnx/src/pass_level2/torch_flip.cpp b/tools/pnnx/src/pass_level2/torch_flip.cpp index 01ba7524b..862a58e12 100644 --- a/tools/pnnx/src/pass_level2/torch_flip.cpp +++ b/tools/pnnx/src/pass_level2/torch_flip.cpp @@ -55,10 +55,7 @@ pnnx.Output output 1 0 out int step = captured_params.at("steps").i; if (axis == 0 && start == -1 && end == INT_MIN + 1 && step == -1) - { - fprintf(stderr, "aaa %d %d %d\n", start, end, step); return true; - } } else // if (captured_params.at("axes").type == 5) { @@ -70,14 +67,10 @@ pnnx.Output output 1 0 out for (size_t i = 0; i < axes.size(); i++) { if (starts[i] != -1 || ends[i] != INT_MIN + 1 || steps[i] != -1) - { - fprintf(stderr, "%d %d %d\n", starts[i], ends[i], steps[i]); return false; - } } } - fprintf(stderr, "bbb\n"); return true; } diff --git a/tools/pnnx/tests/onnx/test_torch_flip.py b/tools/pnnx/tests/onnx/test_torch_flip.py index 8fcaecd55..15b20bb77 100644 --- a/tools/pnnx/tests/onnx/test_torch_flip.py +++ b/tools/pnnx/tests/onnx/test_torch_flip.py @@ -4,6 +4,7 @@ import torch import torch.nn as nn import torch.nn.functional as F +from packaging import version class Model(nn.Module): def __init__(self): @@ -44,6 +45,9 @@ class Model(nn.Module): return x0, y0, y1, y2, z0, z1, z2, z3, z4, z5, z6, w0, w1, w2, w3, w4, w5, w6, w7, w8, w9, w10, w11, w12, w13, w14 def test(): + if version.parse(torch.__version__) < version.parse('1.12'): + return True + net = Model() net.eval()