fix: 改正导出中的笔误

Signed-off-by: YdrMaster <ydrml@hotmail.com>

Makefile

@@ -11,7 +11,7 @@ FORMAT_ORIGIN ?=
 DOCKER_NAME ?= infinitensor
 DOCKER_IMAGE_NAME ?= infinitensor
 DOCKER_FILE ?= infinitensor_ubuntu_22.04.dockerfile
-DOCKER_RUN_OPTION ?= 
+DOCKER_RUN_OPTION ?=
 
 # CUDA option.
 ifeq ($(CUDA), ON)
@@ -58,7 +58,7 @@ test-api:
 	@echo
 	python3 pyinfinitensor/tests/test_api.py
 
-docker-build: 
+docker-build:
 	docker build -f scripts/dockerfile/$(DOCKER_FILE) -t $(DOCKER_NAME) .
 
 docker-run:
@@ -69,5 +69,3 @@ docker-start:
 
 docker-exec:
 	docker exec -it $(DOCKER_IMAGE_NAME) bash
-
-

RefactorGraph

@@ -1 +1 @@
-Subproject commit 7c08ec3d2589de0e5d4c9c9bba805c12e8271309
+Subproject commit b26483a39c1d4813a52718d6c9f793e4f1e349a7

pyinfinitensor/src/pyinfinitensor/onnx.py

@@ -112,15 +112,11 @@ def build_onnx(grpah_name: str, graph: backend.Graph) -> ModelProto:
         edges[name] = make_tensor_value_info(name, data_type, shape)
 
     global_inputs = [
-        edges.pop(name)
-        if name in edges
-        else make_tensor_value_info(name, TensorProto.UNDEFINED, None)
+        edges.pop(name, make_tensor_value_info(name, TensorProto.UNDEFINED, None))
         for name in node_export.global_inputs()
     ]
     global_outputs = [
-        edges.pop(name)
-        if name in edges
-        else make_tensor_value_info(name, TensorProto.UNDEFINED, None)
+        edges.pop(name, make_tensor_value_info(name, TensorProto.UNDEFINED, None))
         for name in node_export.global_outputs()
     ]
     value_info = list(edges.values())

src/ffi/ffi_infinitensor.cc

@@ -81,7 +81,7 @@ class NodeExport {
     }
 
     std::vector<Name> globalOutputs() const {
-        auto outputs = _it.globalInputs();
+        auto outputs = _it.globalOutputs();
         std::vector<Name> ans(outputs.size());
         std::transform(
             outputs.begin(), outputs.end(), ans.begin(),