From d7166de5a4c8a00246ea75d3860d30a6c7b6b460 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=A1=82=E5=86=9B?= <huangguijun@huawei.com>
Date: Wed, 29 Jun 2022 00:50:25 +0800
Subject: [PATCH] add file constant parser st

---
 .../origin_models/file_constant_input.bin     | Bin 0 -> 48 bytes
 .../origin_models/onnx_file_const_type.onnx   |  23 ++++++++++++++++++
 tests/st/testcase/test_onnx_parser.cc         |  18 ++++++++++++++
 3 files changed, 41 insertions(+)
 create mode 100644 tests/st/testcase/origin_models/file_constant_input.bin
 create mode 100644 tests/st/testcase/origin_models/onnx_file_const_type.onnx

diff --git a/tests/st/testcase/origin_models/file_constant_input.bin b/tests/st/testcase/origin_models/file_constant_input.bin
new file mode 100644
index 0000000000000000000000000000000000000000..7e619c9c269f0c4b725432b6197a0ffad9b56b53
GIT binary patch
literal 48
bcmZQzU|?VbVkRJF24WT<W(8spz$Oj=0<Zud

literal 0
HcmV?d00001

diff --git a/tests/st/testcase/origin_models/onnx_file_const_type.onnx b/tests/st/testcase/origin_models/onnx_file_const_type.onnx
new file mode 100644
index 0000000..cce42bb
--- /dev/null
+++ b/tests/st/testcase/origin_models/onnx_file_const_type.onnx
@@ -0,0 +1,23 @@
+:¢
+
+
x
+
y
z"Add
+test_graph*aBfile_constant_input.binj#
+locationfile_constant_input.binj
+offset
0j
+length24p
*bBfile_constant_input.binj#
+locationfile_constant_input.binj
+offset24j
+length24p
Z
+
x
+
+
+Z
+
y
+
+
+b
+
z
+
+
+B
\ No newline at end of file
diff --git a/tests/st/testcase/test_onnx_parser.cc b/tests/st/testcase/test_onnx_parser.cc
index 459abed..498706a 100644
--- a/tests/st/testcase/test_onnx_parser.cc
+++ b/tests/st/testcase/test_onnx_parser.cc
@@ -88,6 +88,7 @@ void STestOnnxParser::RegisterCustomOp() {
   REGISTER_CUSTOM_OP("Add")
   .FrameworkType(domi::ONNX)
       .OriginOpType("ai.onnx::11::Add")
+      .OriginOpType("ai.onnx::17::Add")
       .ParseParamsFn(ParseParams);
 
   REGISTER_CUSTOM_OP("Identity")
@@ -174,4 +175,21 @@ TEST_F(STestOnnxParser, onnx_parser_const_data_type) {
   EXPECT_EQ(ret, GRAPH_SUCCESS);
 }
 
+REG_OP(FileConstant)
+    .OUTPUT(y, TensorType(DT_INT32))
+    .REQUIRED_ATTR(file_id, String)
+    .REQUIRED_ATTR(shape, ListInt)
+    .REQUIRED_ATTR(dtype, Type)
+    .OP_END_FACTORY_REG(FileConstant)
+
+TEST_F(STestOnnxParser, onnx_parser_file_const_data_type) {
+  std::string case_dir = __FILE__;
+  case_dir = case_dir.substr(0, case_dir.find_last_of("/"));
+  std::string model_file = case_dir + "/origin_models/onnx_file_const_type.onnx";
+  std::map<ge::AscendString, ge::AscendString> parser_params;
+  ge::Graph graph;
+  auto ret = ge::aclgrphParseONNX(model_file.c_str(), parser_params, graph);
+  EXPECT_EQ(ret, GRAPH_SUCCESS);
+}
+
 } // namespace ge