提交 77d07df4 编写于 作者: Z zhupengyang 提交者: GitHub

[NPU] remove empty shape (#3687)

上级 3761dbf7
...@@ -120,7 +120,9 @@ int SubgraphEngine::BuildDeviceProgram() { ...@@ -120,7 +120,9 @@ int SubgraphEngine::BuildDeviceProgram() {
return subgraph::FAILED; return subgraph::FAILED;
} }
auto device_program = std::make_shared<device_program_t>(device_client); auto device_program = std::make_shared<device_program_t>(device_client);
device_program_map_[inputs_shape_] = device_program; if (!inputs_shape_.empty()) {
device_program_map_[inputs_shape_] = device_program;
}
// Query and check the dimensions of valid input and output tensors // Query and check the dimensions of valid input and output tensors
std::vector<hiai::TensorDimension> device_idims, device_odims; std::vector<hiai::TensorDimension> device_idims, device_odims;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册