File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -193,7 +193,7 @@ nki builtin.isa.activation
193193 (scale : Sum Immediate Access := .inl $ .float 1 .0 ) -- This also can accept a tensor
194194 (reduce_op : Option AluOp := none)
195195 (reduce_res : Option Access := none)
196- (reduce_cmd : AccumCmd := .Idle )
196+ (reduce_cmd : AccumCmd := .ZeroAccumulate )
197197 (mask : Option Immediate := none)
198198 (name : Option String := none) := do
199199 if mask.isSome then
@@ -220,7 +220,6 @@ nki builtin.isa.activation_reduce
220220 -- kwargs
221221 (reduce_op : Option AluOp := none)
222222 (reduce_res : Option Access := none)
223- (reduce_cmd : AccumCmd := .Idle)
224223 (bias : Option Access := none )
225224 (scale : Sum Immediate Access := .inl $ .float 1 .0 )
226225 (mask : Option Immediate := none)
@@ -239,7 +238,7 @@ nki builtin.isa.activation_reduce
239238 bias := bias.map .abstract,
240239 reduceOp := reduce_op,
241240 reduceRes := reduce_res.map .abstract,
242- accumulatorCmd := reduce_cmd ,
241+ accumulatorCmd := .ZeroAccumulate ,
243242 dtype := dst.tensor.dtype,
244243 }) name
245244 return .none
You can’t perform that action at this time.
0 commit comments