From 108722c7feea8fab32c760345e76701fee686ea5 Mon Sep 17 00:00:00 2001 From: xiguadong Date: Thu, 4 Dec 2025 13:56:12 +0800 Subject: [PATCH 1/2] fix set outputs names --- axengine/_axclrt.py | 4 +++- axengine/_axe.py | 5 ++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/axengine/_axclrt.py b/axengine/_axclrt.py index e07fa59..109d329 100644 --- a/axengine/_axclrt.py +++ b/axengine/_axclrt.py @@ -372,8 +372,10 @@ def run( # get output outputs = [] + origin_output_names = [_o.name for _o in self.get_outputs(shape_group)] + outputs_ranks = [output_names.index(_on) for _on in origin_output_names] if 0 == ret: - for i in range(len(self.get_outputs(shape_group))): + for i in outputs_ranks: ret = axclrt_lib.axclrtEngineGetOutputBufferByIndex(self._io[0], i, dev_prt, dev_size) if 0 != ret: raise RuntimeError(f"axclrtEngineGetOutputBufferByIndex failed for output {i}.") diff --git a/axengine/_axe.py b/axengine/_axe.py index 6bfb431..aa027cf 100644 --- a/axengine/_axe.py +++ b/axengine/_axe.py @@ -395,8 +395,11 @@ def run( # flush output outputs = [] + origin_output_names = [_o.name for _o in self.get_outputs(shape_group)] + outputs_ranks = [output_names.index(_on) for _on in origin_output_names] + if 0 == ret: - for i in range(len(self.get_outputs(shape_group))): + for i in outputs_ranks: sys_lib.AX_SYS_MinvalidateCache( self._io[0].pOutputs[i].phyAddr, self._io[0].pOutputs[i].pVirAddr, From 597ba734db3c695e1108e03ce9dda5574939e458 Mon Sep 17 00:00:00 2001 From: xiguadong Date: Thu, 4 Dec 2025 13:58:51 +0800 Subject: [PATCH 2/2] fix typo --- axengine/_axe.py | 1 - 1 file changed, 1 deletion(-) diff --git a/axengine/_axe.py b/axengine/_axe.py index aa027cf..f6deffb 100644 --- a/axengine/_axe.py +++ b/axengine/_axe.py @@ -397,7 +397,6 @@ def run( outputs = [] origin_output_names = [_o.name for _o in self.get_outputs(shape_group)] outputs_ranks = [output_names.index(_on) for _on in origin_output_names] - if 0 == ret: for i in outputs_ranks: sys_lib.AX_SYS_MinvalidateCache(