Skip to content

Commit afad30b

Browse files
ChenxiQwangqiankun13
authored andcommitted
add custom op GmmSwigluQuantWeightNZTensorList
Signed-off-by: QianChenxi <[email protected]> [Kernel] add custom op DispatchGmmCombineDecode Signed-off-by: wangqiankun13 <[email protected]>
1 parent 136ea9f commit afad30b

File tree

78 files changed

+15467
-7
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

78 files changed

+15467
-7
lines changed

.github/workflows/release_whl.yml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,12 @@ jobs:
9595
--exclude libge_common_base.so \
9696
--exclude libc10.so \
9797
--exclude libc_sec.so \
98+
--exclude libopapi.so \
99+
--exclude libnnopbase.so \
100+
--exclude libprofapi.so \
101+
--exclude libgraph_base.so \
102+
--exclude libgraph.so \
103+
--exclude libexe_graph.so \
98104
--exclude "libascend*.so" \
99105
--exclude "libtorch*.so" \
100106
--exclude "liberror_manager.so"

.gitsubmodules

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
[submodule "csrc/third_party/catlass"]
2+
path = csrc/third_party/catlass
3+
url = https://gitee.com/ascend/catlass.git
4+
branch = catlass-v1-stable

.pre-commit-config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ repos:
1212
- id: codespell
1313
args: [
1414
--toml, pyproject.toml,
15-
'--skip', 'tests/e2e/multicard/test_torchair_graph_mode.py,csrc/mla_preprocess/**,tests/prompts/**,./benchmarks/sonnet.txt,*tests/lora/data/**,build/**,./vllm_ascend.egg-info/**,.github/**,typos.toml',
15+
'--skip', 'tests/e2e/multicard/test_torchair_graph_mode.py,csrc/**,tests/prompts/**,./benchmarks/sonnet.txt,*tests/lora/data/**,build/**,./vllm_ascend.egg-info/**,.github/**,typos.toml',
1616
'-L', 'CANN,cann,NNAL,nnal,ASCEND,ascend,EnQue,CopyIn,ArchType,AND'
1717
]
1818
additional_dependencies:
@@ -37,7 +37,7 @@ repos:
3737
- id: typos
3838
args: [
3939
"--force-exclude",
40-
"--exclude", "csrc/mla_preprocess/**"
40+
"--exclude", "csrc/**"
4141
]
4242
- repo: https://github.com/PyCQA/isort
4343
rev: 6.0.1

CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ set(
8181
${TORCH_NPU_INCLUDE_DIRS}
8282
${ASCEND_HOME_PATH}/include
8383
${ASCEND_HOME_PATH}/aarch64-linux/include/experiment/platform
84+
${ASCEND_HOME_PATH}/x86_64-linux/include/experiment/platform
8485
)
8586

8687
pybind11_add_module(vllm_ascend_C ${VLLM_ASCEND_SRC})

0 commit comments

Comments
 (0)