github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-10 13:30:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Inference/Refactor] Delete Duplicated code and refactor vec_copy utils and reduce utils (#5593)

Browse Source 
* delete duplicated code and refactor vec_copy utils and reduce utils

* delete unused header file
This commit is contained in:
傅剑寒
2024-04-15 10:57:51 +08:00
committed by GitHub
parent a21912339a
commit d4cb023b62
16 changed files with 161 additions and 345 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
extensions/csrc/__init__.py
View File

@@ -1,11 +0,0 @@
from .layer_norm import MixedFusedLayerNorm as LayerNorm
from .multihead_attention import MultiHeadAttention
from .scaled_softmax import AttnMaskType, FusedScaleMaskSoftmax, ScaledUpperTriangMaskedSoftmax
__all__ = [
"LayerNorm",
"MultiHeadAttention",
"FusedScaleMaskSoftmax",
"ScaledUpperTriangMaskedSoftmax",
"AttnMaskType",
]