github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
pull/5568/head
pre-commit-ci[bot] 2024-04-08 08:41:07 +00:00
parent ce9401ad52
commit d78817539e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
extensions/csrc/cuda/pybind/inference.cpp
View File

@ -80,6 +80,5 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("fused_add_rms_layernorm", &fused_add_rms_layernorm, m.def("fused_add_rms_layernorm", &fused_add_rms_layernorm,
"In-place fused Add and RMS Normalization."); "In-place fused Add and RMS Normalization.");
m.def("get_cos_and_sin", &get_cos_and_sin, m.def("get_cos_and_sin", &get_cos_and_sin, "Get cos and sin from the cache.");
"Get cos and sin from the cache.");
} }