Skip to content

Commit d6fd7fe

Browse files
xzliu-optDongxu-H
andauthored
[KernelGen][Hygon] Add max_pool3d_with_indices Hygon specialized operator (#140)
Co-authored-by: Dongxu-H <dxhan@baai.ac.cn>
1 parent c8a16fa commit d6fd7fe

2 files changed

Lines changed: 510 additions & 0 deletions

File tree

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
from .isin import isin
3333
from .matmul_bf16 import matmul_bf16
3434
from .matmul_int8 import matmul_int8
35+
from .max_pool3d_with_indices import max_pool3d_backward, max_pool3d_with_indices
3536
from .median import median_dim, median_dim_values
3637
from .mm import mm
3738
from .mul import mul, mul_
@@ -75,6 +76,8 @@
7576
"isin",
7677
"matmul_bf16",
7778
"matmul_int8",
79+
"max_pool3d_backward",
80+
"max_pool3d_with_indices",
7881
"median_dim",
7982
"median_dim_values",
8083
"mul",

0 commit comments

Comments
 (0)