From a2e1867843788fe2fb65c6f1aeffa5b5e06aac66 Mon Sep 17 00:00:00 2001 From: zjgemi Date: Wed, 29 May 2024 13:36:59 +0800 Subject: [PATCH] fix: pass on sliced big parameter for sliced super OP Signed-off-by: zjgemi --- src/dflow/step.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dflow/step.py b/src/dflow/step.py index e271d44e..8c627cc7 100644 --- a/src/dflow/step.py +++ b/src/dflow/step.py @@ -2288,8 +2288,8 @@ def add_slices(templ: OPTemplate, slices: Slices, layer=0): for step in steps: for par in list(step.inputs.parameters.values()): # input parameter referring to sliced input parameter - if getattr(par, "value", None) is \ - templ.inputs.parameters[name]: + if getattr(par, "value", None) is templ.inputs.parameters[ + name] or par.source is templ.inputs.parameters[name]: templ.inputs.parameters["dflow_skip_slice_input"] = \ InputParameter(value=0) if hasattr(step.template, "set_skip_slice_input"):