Skip to content

Commit cd36478

Browse files
authored
zq/fix sqrt (DeepLink-org#908)
* fix sqrt * fix device_config
1 parent 536513b commit cd36478

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

impl/ascend/convert_config.yaml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,11 +161,10 @@
161161
dtype: (float64)->float32
162162

163163
- diopiSqrt:
164-
# 这里转为float64的原因有三,SqrtInp同理:1.float16会产生NAN数据,只有转化为float64才能解决 2.float32数据会在某些用例下产生华为算子的E39999错误 3.Sqrt底层算子不支持int类型
165-
dtype: (int8, int16, int32, int64, uint8, uint16, uint32, uint64, bool, float16, float32)->float64, (complex32)->complex64
164+
dtype: (int8, int16, int32, int64, uint8, uint16, uint32, uint64, bool)->float32
166165

167166
- diopiSqrtInp:
168-
dtype: (int8, int16, int32, int64, uint8, uint16, uint32, uint64, bool, float16, float32)->float64, (complex32)->complex64
167+
dtype: (int8, int16, int32, int64, uint8, uint16, uint32, uint64, bool)->float32
169168

170169
- diopiLayerNorm:
171170
dtype: (float64)->float32

0 commit comments

Comments
 (0)