githubmirror/ColossalAI | Gitverse

/

/

Обзор ДокументацияВойти

/

/

Аналитика

Безопасность

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[kernel] fixed repeated loading of kernels (#2549)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

multi_head_attn.py

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

scaled_masked_softmax.py

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

scaled_upper_triangle_masked_softmax.py

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

README.md