Skip to content

Commit 464ecb2

Browse files
committed
[Ops][Refactor] Strip to arch35-only, rename to FFNLinear/aclnnFFNLinear with delivery cleanup
Signed-off-by: chenchris2 <1349418798@qq.com>
1 parent 4025e8f commit 464ecb2

134 files changed

Lines changed: 705 additions & 10108 deletions

File tree

Some content is hidden

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

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ __pycache__/
3636
.Python
3737
build/
3838
build_out/
39+
build_cxx/
3940
!csrc/cmake/third_party/build/
4041
!csrc/cmake/third_party/build/modules/
4142
!csrc/cmake/third_party/build/modules/patch/

csrc/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,7 @@ list(APPEND OP_DIR_LIST ${CMAKE_CURRENT_SOURCE_DIR}/moe/moe_init_routing_v2)
300300
list(APPEND OP_DIR_LIST ${CMAKE_CURRENT_SOURCE_DIR}/moe/moe_token_unpermute_with_ep_grad)
301301
add_subdirectory(ffn)
302302
list(APPEND OP_LIST "ffn")
303-
list(APPEND OP_DIR_LIST ${CMAKE_CURRENT_SOURCE_DIR}/ffn/ffn)
303+
list(APPEND OP_DIR_LIST ${CMAKE_CURRENT_SOURCE_DIR}/ffn/ffn_linear)
304304
add_subdirectory(attention)
305305
list(APPEND OP_LIST ${COMPILED_OPS})
306306
list(REMOVE_DUPLICATES OP_LIST)

csrc/build_aclnn.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ elif [[ "$SOC_VERSION" =~ ^ascend950 ]]; then
225225
"k2q_csr"
226226
"sparse_attention_score"
227227
"mla_prolog_v3"
228-
"ffn"
228+
"ffn_linear"
229229
)
230230

231231
CUSTOM_OPS=$(IFS=';'; echo "${CUSTOM_OPS_ARRAY[*]}")

csrc/ffn/README.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ export SOC_VERSION=ascend950
2020
pip install -e .
2121

2222
# 路径 B(单算子,调试用):
23-
cd csrc && bash build.sh --pkg --ops=ffn --soc=ascend950 -j8
23+
cd csrc && bash build.sh --pkg --ops=ffn_linear --soc=ascend950 -j8
2424
```
2525

2626
## 调用
@@ -48,6 +48,9 @@ Meta kernel 已注册(`torch_binding_meta.cpp`),支持 symbolic tracing / ACLGra
4848

4949
## 已知限制
5050

51+
- 算子命名与官方 ops-transformer `FFN` 完全隔离:op type `FFNLinear`、aclnn 接口
52+
`aclnnFFNLinear`、kernel 二进制 `ffn_linear`(官方转 catlass 合入 ops-transformer 后,
53+
可按仓内惯例删除自持实现切回官方 API)。
5154
- swiglu 仅支持 Linear 布局 weight1 `[2H,K]`;gelu/silu 支持 canonical `[K,H]`
5255
linear `[N,K]` 双布局;**全方阵(H==K==N)歧义默认 linear**
5356
(PyTorch Linear 惯例,与独立 ops-transformer 仓有意分叉)。

0 commit comments

Comments
 (0)