kernels-community
/

flash-mla

Model card Files Files and versions Community

flash-mla / flash_mla

Ctrl+K

Ctrl+K

3 contributors

History: 4 commits

drbh

fix: adjust sig types

59bdff8 2 months ago

flash_fwd_mla_bf16_sm90.cu

151 Bytes

feat: build flash mla with kernel builder 2 months ago
flash_fwd_mla_fp16_sm90.cu

147 Bytes

feat: build flash mla with kernel builder 2 months ago
flash_fwd_mla_kernel.h

32.5 kB

feat: build flash mla with kernel builder 2 months ago
flash_fwd_mla_metadata.cu

3.55 kB

feat: build flash mla with kernel builder 2 months ago
flash_mla.h

1.81 kB

feat: build flash mla with kernel builder 2 months ago
flash_mla_api.cu

9.38 kB

fix: adjust sig types 2 months ago
named_barrier.h

316 Bytes

feat: build flash mla with kernel builder 2 months ago
softmax.h

9.6 kB

feat: build flash mla with kernel builder 2 months ago
static_switch.h

3.75 kB

feat: build flash mla with kernel builder 2 months ago
utils.h

11.4 kB

feat: build flash mla with kernel builder 2 months ago