diff --git a/csrc/mmdeploy/backend_ops/torchscript/optimizer/ir/subgraph_matcher.cpp b/csrc/mmdeploy/backend_ops/torchscript/optimizer/ir/subgraph_matcher.cpp index d7df0704fc..6f188c5688 100644 --- a/csrc/mmdeploy/backend_ops/torchscript/optimizer/ir/subgraph_matcher.cpp +++ b/csrc/mmdeploy/backend_ops/torchscript/optimizer/ir/subgraph_matcher.cpp @@ -125,7 +125,7 @@ bool SubgraphMatcher::SubgraphMatcherImpl::matchAttributes(const Node* n1, Node* "' did not match:\n", *n1, *n2); return false; } - std::vector n1is, n2is; + std::vector n1is, n2is; std::vector n1fs, n2fs; switch (n1->kindOf(attr_name)) { case AttributeKind::s: diff --git a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/flatten_cls_head.cpp b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/flatten_cls_head.cpp index 5c7082f6ae..73f8965412 100644 --- a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/flatten_cls_head.cpp +++ b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/flatten_cls_head.cpp @@ -67,7 +67,7 @@ static bool matchClsHead(const Match& match, const std::unordered_mapt(Symbol::attr("value")); if (ival.dim() != 0) return false; - auto ival_dataptr = ival.data_ptr(); + auto ival_dataptr = ival.data_ptr(); if (ival_dataptr[0] != 0) return false; } diff --git a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/fuse_select_assign.cpp b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/fuse_select_assign.cpp index b2904b909d..8dc5847753 100644 --- a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/fuse_select_assign.cpp +++ b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/fuse_select_assign.cpp @@ -65,7 +65,7 @@ bool FuseSelectAssign(Node* node, std::unordered_map& param Node* shape = values_map[vmap["reshape_1_shape"]]->node(); auto shape_val = shape->t(Symbol::attr("value")); if (shape_val.dim() != 1) return false; - if (shape_val.data_ptr()[0] != -1) return false; + if (shape_val.data_ptr()[0] != -1) return false; } { @@ -83,7 +83,7 @@ bool FuseSelectAssign(Node* node, std::unordered_map& param Node* gather_inds = values_map[vmap["gather_inds_2"]]->node(); auto inds_val = gather_inds->t(Symbol::attr("value")); if (inds_val.dim() != 0) return false; - if (inds_val.data_ptr()[0] != 0) return false; + if (inds_val.data_ptr()[0] != 0) return false; } { @@ -92,7 +92,7 @@ bool FuseSelectAssign(Node* node, std::unordered_map& param auto start_name = slice->inputs()[1]->debugName(); auto start_val = params[start_name]; if (start_val.dim() != 1) return false; - if (start_val.data_ptr()[0] != 0) return false; + if (start_val.data_ptr()[0] != 0) return false; } // create new node diff --git a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/merge_shape_concate.cpp b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/merge_shape_concate.cpp index 47fc8c205f..ad0551fa1b 100644 --- a/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/merge_shape_concate.cpp +++ b/csrc/mmdeploy/backend_ops/torchscript/optimizer/passes/onnx/merge_shape_concate.cpp @@ -18,7 +18,7 @@ using torch::jit::Value; void MergeShapeConcate(Node* node) { auto inputs = node->inputs(); - std::vector gather_value; + std::vector gather_value; Value* shape_from = nullptr; std::vector node_to_remove{node}; @@ -54,7 +54,7 @@ void MergeShapeConcate(Node* node) { if (!is_kind(constant_node, "onnx::Constant")) return; auto gather_indices_val = constant_node->t(Symbol::attr("value")); - long* data_ptr = gather_indices_val.data_ptr(); + int64_t* data_ptr = gather_indices_val.data_ptr(); if (gather_indices_val.dim() == 0) { gather_value.push_back(data_ptr[0]); } else {