kernels-community
/

flash-mla

Model card Files Files and versions Community

flash-mla / flash_mla

1 contributor

History: 1 commit

drbh

feat: build flash mla with kernel builder

1f83cde 29 days ago

flash_fwd_mla_bf16_sm90.cu

151 Bytes

feat: build flash mla with kernel builder 29 days ago
flash_fwd_mla_fp16_sm90.cu

147 Bytes

feat: build flash mla with kernel builder 29 days ago
flash_fwd_mla_kernel.h

32.5 kB

feat: build flash mla with kernel builder 29 days ago
flash_fwd_mla_metadata.cu

3.55 kB

feat: build flash mla with kernel builder 29 days ago
flash_mla.h

1.81 kB

feat: build flash mla with kernel builder 29 days ago
flash_mla_api.cu

10.7 kB

feat: build flash mla with kernel builder 29 days ago
named_barrier.h

316 Bytes

feat: build flash mla with kernel builder 29 days ago
softmax.h

9.6 kB

feat: build flash mla with kernel builder 29 days ago
static_switch.h

3.75 kB

feat: build flash mla with kernel builder 29 days ago
utils.h

11.4 kB

feat: build flash mla with kernel builder 29 days ago