Browse Source

pnnx pytorch 1.13 (#4314)

tags/20221128
nihui GitHub 3 years ago
parent
commit
fdf129fcd7
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 49 additions and 0 deletions
  1. +3
    -0
      .github/workflows/pnnx.yml
  2. +23
    -0
      tools/pnnx/src/pass_level2/F_upsample_nearest.cpp
  3. +23
    -0
      tools/pnnx/src/pass_level2/torch_einsum.cpp

+ 3
- 0
.github/workflows/pnnx.yml View File

@@ -41,6 +41,9 @@ jobs:
- torch-version: 1.12.0
torchvision-version: 0.13.0

- torch-version: 1.13.0
torchvision-version: 0.14.0

steps:
- uses: actions/checkout@v3
with:


+ 23
- 0
tools/pnnx/src/pass_level2/F_upsample_nearest.cpp View File

@@ -63,6 +63,29 @@ pnnx.Output output 1 0 out

REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(F_upsample_nearest_1, 10)

class F_upsample_nearest_1_1 : public GraphRewriterPass
{
public:
const char* match_pattern_graph() const
{
return R"PNNXIR(7767517
5 4
pnnx.Input input_0 0 1 input
pnnx.Input input_1 0 1 size
prim::Constant op_0 0 1 scale_factor value=None
aten::upsample_nearest2d op_1 3 1 input size scale_factor out
pnnx.Output output 1 0 out
)PNNXIR";
}

const char* type_str() const
{
return "F.upsample_nearest";
}
};

REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(F_upsample_nearest_1_1, 10)

class F_upsample_nearest_2 : public GraphRewriterPass
{
public:


+ 23
- 0
tools/pnnx/src/pass_level2/torch_einsum.cpp View File

@@ -38,4 +38,27 @@ pnnx.Output output 1 0 out

REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(torch_einsum, 20)

class torch_einsum_1 : public GraphRewriterPass
{
public:
const char* match_pattern_graph() const
{
return R"PNNXIR(7767517
5 4
pnnx.Input input_0 0 1 equation
pnnx.Input input_1 0 1 operands
prim::Constant op_0 0 1 path value=None
aten::einsum op_1 3 1 equation operands path out
pnnx.Output output 1 0 out
)PNNXIR";
}

const char* type_str() const
{
return "torch.einsum";
}
};

REGISTER_GLOBAL_PNNX_GRAPH_REWRITER_PASS(torch_einsum_1, 20)

} // namespace pnnx

Loading…
Cancel
Save