Skip to content

Commit 011b74b

Browse files
authored
zq/fix repeat device config (DeepLink-org#863)
fix repeat device config
1 parent 2acdecd commit 011b74b

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

impl/ascend/device_configs.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1365,15 +1365,12 @@
13651365
),
13661366
),
13671367

1368+
# Ascend Not support Tile shape max than 8 on dynamic rank case.
13681369
'repeat': dict( # llm used
13691370
name=['repeat'],
1370-
tensor_para=dict(
1371-
args=[
1372-
{
1373-
"ins": ['input'],
1374-
"shape": [Skip((4, 2, 3, 5)), Skip(()), Skip((0,)), Skip((12, 0)), Skip((4, 0, 9))],
1375-
},
1376-
]
1371+
para=dict(
1372+
repeats=
1373+
[Skip((3, 4, 6, 3, 5))],
13771374
),
13781375
),
13791376

0 commit comments

Comments
 (0)