Add zero on increment support
parent
3ed0739bbe
commit
fc12e72906
|
@ -172,6 +172,7 @@ async def cascade(
|
|||
|
||||
symbol: str,
|
||||
func_name: str,
|
||||
zero_on_step: bool = False,
|
||||
|
||||
loglevel: Optional[str] = None,
|
||||
|
||||
|
@ -233,6 +234,11 @@ async def cascade(
|
|||
)
|
||||
|
||||
cs, index = await n.start(fsp_target)
|
||||
|
||||
if zero_on_step:
|
||||
last = dst.array[-1:]
|
||||
zeroed = np.zeros(last.shape, dtype=last.dtype)
|
||||
|
||||
await ctx.started(index)
|
||||
profiler(f'{func_name}: fsp up')
|
||||
|
||||
|
@ -264,6 +270,9 @@ async def cascade(
|
|||
# TODO: some signals, like vlm should be reset to
|
||||
# zero every step.
|
||||
last = array[-1:].copy()
|
||||
if zero_on_step:
|
||||
last = zeroed
|
||||
|
||||
dst.push(last)
|
||||
last_len = new_len
|
||||
|
||||
|
|
Loading…
Reference in New Issue