summaryrefslogtreecommitdiff
path: root/candle-metal-kernels/src
ModeNameSize
-rw-r--r--affine.metal3206logplain
-rw-r--r--binary.metal3624logplain
-rw-r--r--cast.metal4230logplain
-rw-r--r--conv.metal18042logplain
-rw-r--r--fill.metal1045logplain
-rw-r--r--indexing.metal8739logplain
-rw-r--r--lib.rs84336logplain
-rw-r--r--libMetalFlashAttention.metallib116184logplain
-rw-r--r--mlx_gemm.metal43718logplain
-rw-r--r--quantized.metal178656logplain
-rw-r--r--random.metal7118logplain
-rw-r--r--reduce.metal19591logplain
-rw-r--r--scaled_dot_product_attention.metal37401logplain
-rw-r--r--sort.metal3214logplain
-rw-r--r--ternary.metal3539logplain
-rw-r--r--tests.rs67057logplain
-rw-r--r--unary.metal5302logplain
-rw-r--r--utils.rs6014logplain