GitHub - charlotteroes/training_lib: for cuda training lib

python3 setup.py install

python3 gating.py

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
.gitignore		.gitignore
README.md		README.md
bf16_diff.py		bf16_diff.py
build.sh		build.sh
gating.py		gating.py
gating_lib.h		gating_lib.h
libcuda_static_lib.a		libcuda_static_lib.a
librope_cuda_static_lib.a		librope_cuda_static_lib.a
mla_rope.cpp		mla_rope.cpp
mla_rope_lib.h		mla_rope_lib.h
mla_rope_train.py		mla_rope_train.py
ppl_deepseek_kernel.cpp		ppl_deepseek_kernel.cpp
run_test.sh		run_test.sh
setup.py		setup.py

Provide feedback