baracuda-kernels

This version has 19 feature flags, 2 of them enabled by default.

default

• fa2 (default)
• sm80 (default)

fa2 (default)

• baracuda-kernels-sys/fa2 (default)

sm80 (default)

• baracuda-cutlass/sm80 (default)
• baracuda-kernels-sys/sm80 (default)

awq

• baracuda-kernels-sys/awq

bnb_nf4

• baracuda-kernels-sys/bnb_nf4

cudnn

• baracuda-kernels-sys/cudnn

flashinfer

• baracuda-kernels-sys/flashinfer

mamba

• baracuda-kernels-sys/mamba

marlin

• baracuda-kernels-sys/marlin

megatron_tp

• baracuda-megatron/half-crate
• dep:baracuda-megatron

mhc

• baracuda-kernels-sys/mhc

nvshmem

• dep:baracuda-nvshmem
• dep:baracuda-nvshmem-sys

optim

• baracuda-optim/sm80
• dep:baracuda-optim

ozimmu

• baracuda-cutlass/ozimmu

ring_attention

• baracuda-kernels-sys/ring_attention
• dep:baracuda-nccl
• dep:baracuda-nccl-sys

sm89

• baracuda-kernels-sys/sm89

sm90a

• baracuda-cutlass/sm90a
• baracuda-kernels-sys/sm90a

tensor_engine

• baracuda-transformer-engine/sm80
• dep:baracuda-transformer-engine

xformers_blocksparse

• baracuda-kernels-sys/xformers_blocksparse

xformers_sparse24

• baracuda-kernels-sys/xformers_sparse24