Skip to content

Commit cefb52b

Browse files
authored
[KMCompiler][Ascend][MetaX] Add nansum operator with triton kernel (flagos-ai#5377)
1 parent 32a4654 commit cefb52b

5 files changed

Lines changed: 1410 additions & 0 deletions

File tree

src/flag_gems/runtime/backend/_ascend/ops/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
from .min import min, min_dim
6868
from .mm import mm, mm_out
6969
from .multinomial import multinomial
70+
from .nansum import nansum, nansum_out
7071
from .nonzero_static import nonzero_static, nonzero_static_out
7172
from .ones import ones
7273
from .ones_like import ones_like
@@ -178,6 +179,8 @@
178179
"mm",
179180
"mm_out",
180181
"multinomial",
182+
"nansum",
183+
"nansum_out",
181184
"nonzero_static",
182185
"nonzero_static_out",
183186
"normed_cumsum",

0 commit comments

Comments
 (0)