Skip to content

Commit e020bcf

Browse files
committed
Fix keepdims type in fuse mhsa pass
1 parent ff6295d commit e020bcf

File tree

1 file changed

+1
-1
lines changed
  • Deeploy/Targets/MemPool/TopologyOptimizationPasses

1 file changed

+1
-1
lines changed

Deeploy/Targets/MemPool/TopologyOptimizationPasses/Passes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ def get_constant_input_or_zeros(n: gs.Node, shape):
289289
name = name + "_sum",
290290
attrs = {
291291
'axes': [1],
292-
"keepdims": "0"
292+
"keepdims": 0
293293
})
294294

295295
mhsa_out[0].shape = [_output.shape[0]] + [int(H)] + _output.shape[1:]

0 commit comments

Comments
 (0)