feat: add npugraph_ex support for MHC expand and head mix-ops-tilelang-AtomGit

Star257

Star257

Open

feat: add npugraph_ex support for MHC expand and head mix #62

wangjiaxingAdd on 6 days agoMerged fromgcw_Co0B5I37/ops-tilelang： feat/mhc-expand-head-aclgraph into cann/ops-tilelang： master

feat: add npugraph_ex support for MHC expand and head mix #62

Open

wangjiaxingAdd on 6 days ago

Merged fromgcw_Co0B5I37/ops-tilelang feat/mhc-expand-head-aclgraphintocann/ops-tilelang master

Total: 8 files changed+952-66

Asrc/cann_ops_tilelang/common/compile_utils.py+49-0

Msrc/cann_ops_tilelang/mhc/__init__.py+4-2

Msrc/cann_ops_tilelang/mhc/expand.py+85-26

Msrc/cann_ops_tilelang/mhc/head_compute_mix.py+178-38

Atests/mhc/test_expand_compile.py+95-0

Atests/mhc/test_head_compute_mix_compile.py+157-0

Atests/mhc/test_mhc_graph_configuration.py+163-0

Atests/mhc/test_mhc_graph_utils.py+221-0

File content is under review, please try again later