Browse Source

fix crop bug and eliminate Tensor.clone (#4416)

* fix crop bug and eliminate Tensor.clone

* fix crop slice bug on msvc

* fix slice bug on msvc
tags/20230223
Yoh GitHub 3 years ago
parent
commit
0fffefd3c9
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
  1. +4
    -4
      tools/pnnx/src/ir.cpp
  2. +1
    -1
      tools/pnnx/src/pass_ncnn/eliminate_noop.cpp

+ 4
- 4
tools/pnnx/src/ir.cpp View File

@@ -183,8 +183,8 @@ Parameter::Parameter(const torch::jit::Node* value_node)
{
type = 2;
int64_t i64 = value_node->i(torch::jit::attr::value);
if (i64 == LONG_MAX) i64 = INT_MAX;
if (i64 == LONG_MIN) i64 = INT_MIN;
if (i64 == std::numeric_limits<int64_t>::max()) i64 = INT_MAX;
if (i64 == std::numeric_limits<int64_t>::max()) i64 = INT_MIN;
i = (int)i64;
break;
}
@@ -210,8 +210,8 @@ Parameter::Parameter(const torch::jit::Node* value_node)
{
type = 2;
int64_t i64 = t.item<int64_t>();
if (i64 == LONG_MAX) i64 = INT_MAX;
if (i64 == LONG_MIN) i64 = INT_MIN;
if (i64 == std::numeric_limits<int64_t>::max()) i64 = INT_MAX;
if (i64 == std::numeric_limits<int64_t>::max()) i64 = INT_MIN;
i = (int)i64;
}
else if (t.scalar_type() == c10::ScalarType::Int)


+ 1
- 1
tools/pnnx/src/pass_ncnn/eliminate_noop.cpp View File

@@ -30,7 +30,7 @@ void eliminate_noop(Graph& graph)
{
Operator* op = graph.ops[i];

if (op->type != "Noop")
if (op->type != "Noop" && op->type != "Tensor.clone")
continue;

need_eliminate = true;


Loading…
Cancel
Save