diff --git a/.circleci/config.yml b/.circleci/config.yml index 48fda88..3863f72 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -38,7 +38,7 @@ jobs: - run: name: Run End-To-End Tests command: | - sudo pip install -q onnx + sudo pip install -q -e ./ONNF/third_party/onnx cd ONNF/build cmake --build . --target run-onnx-backend-test - run: diff --git a/src/builder/frontend_dialect_transformer.cpp b/src/builder/frontend_dialect_transformer.cpp index cd23e8c..0efca22 100644 --- a/src/builder/frontend_dialect_transformer.cpp +++ b/src/builder/frontend_dialect_transformer.cpp @@ -189,8 +189,9 @@ private: } } - mlir::Type elementType = - convertONNXTypeToMLIRType(input.type().tensor_type().elem_type()); + auto elementOnnxType = + (onnx::TensorProto_DataType)input.type().tensor_type().elem_type(); + mlir::Type elementType = convertONNXTypeToMLIRType(elementOnnxType); llvm::ArrayRef tensor_dims(dims.data(), dims.size()); arg_types.emplace_back( mlir::RankedTensorType::get(tensor_dims, elementType)); diff --git a/third_party/onnx b/third_party/onnx index 1439eab..553df22 160000 --- a/third_party/onnx +++ b/third_party/onnx @@ -1 +1 @@ -Subproject commit 1439eab5542c625bb3da49860f0cd68c3eafdc18 +Subproject commit 553df22c67bee5f0fe6599cff60f1afc6748c635