Skip to content

Commit 20ce541

Browse files
authored
zq/update device_configs.py (DeepLink-org#855)
update ../../impl/ascend/device_configs.py
1 parent a76019a commit 20ce541

File tree

1 file changed

+1
-31
lines changed

1 file changed

+1
-31
lines changed

impl/ascend/device_configs.py

Lines changed: 1 addition & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import numpy as np
33
from skip import Skip
44

5-
# topk, normal, norm, nll_loss, gather, fill_, triu llm used
5+
# topk, normal, norm, nll_loss, gather, fill_, triu, bmm, mm llm used
66

77
device_configs = {
88
# temp for 910B
@@ -469,12 +469,6 @@
469469
),
470470
),
471471

472-
'bmm': dict( # llm used
473-
name=['bmm'],
474-
atol=3e-2,
475-
rtol=3e-2,
476-
),
477-
478472
'matmul': dict(
479473
name=['matmul'],
480474
tensor_para=dict(
@@ -938,18 +932,6 @@
938932
),
939933
),
940934

941-
'mm': dict( # llm used
942-
name=['mm'],
943-
atol=2e-2,
944-
rtol=2e-2,
945-
),
946-
947-
'mm_diff_dtype': dict( # llm used
948-
name=['mm'],
949-
atol=2e-2,
950-
rtol=2e-2,
951-
),
952-
953935
'index_fill_scalar': dict(
954936
name=['index_fill'],
955937
tensor_para=dict(
@@ -1134,18 +1116,6 @@
11341116
),
11351117
),
11361118

1137-
'scatter': dict( # llm used
1138-
name=['scatter'],
1139-
tensor_para=dict(
1140-
args=[
1141-
{
1142-
"ins": ['input'],
1143-
"dtype": [Skip(np.float32),Skip(np.float64),Skip(np.float16),Skip(np.int16),Skip(np.int32),Skip(np.int64),Skip(np.uint8),Skip(np.int8),Skip(np.bool_),],
1144-
},
1145-
]
1146-
),
1147-
),
1148-
11491119
'scatter_scalar': dict( # llm used
11501120
name=['scatter'],
11511121
para=dict(

0 commit comments

Comments
 (0)