Kernels:

flashrt
/

masked-mha-runtime

Kernel card Files Files and versions

masked-mha-runtime / csrc

14.4 kB

Ctrl+K

Ctrl+K

1 contributor

History: 2 commits

liangsu9988's picture

Publish GROOT Thor Batch 4 (torch213-cxx11-cu130-aarch64-linux, 6ab0803)

03b8715 verified about 1 month ago

attention_mha_masked.cu

8.69 kB
Publish GROOT N1.7 Thor kernel sources about 2 months ago
attention_mha_masked.cuh

698 Bytes
Publish GROOT N1.7 Thor kernel sources about 2 months ago
attention_seqused_fused.cu

4.28 kB
Publish GROOT Thor Batch 4 (torch213-cxx11-cu130-aarch64-linux, 6ab0803) about 1 month ago
attention_seqused_fused.cuh

699 Bytes
Publish GROOT Thor Batch 4 (torch213-cxx11-cu130-aarch64-linux, 6ab0803) about 1 month ago