Johannes Gäßler e141ce624a Fix FlashAttention debug test, FP32 assert (#7684)

2024-06-01 23:26:10 +02:00

..

template-instances

CUDA: quantized KV support for FA vec (#7527)

2024-06-01 08:44:14 +02:00

acc.cu

…

acc.cuh

…

arange.cu

…

arange.cuh

…

argsort.cu

…

argsort.cuh

…

binbcast.cu

…

binbcast.cuh

…

clamp.cu

…

clamp.cuh

…

common.cuh

…

concat.cu

…

concat.cuh

…

convert.cu

…

convert.cuh

…

cpy.cu

…

cpy.cuh

…

dequantize.cuh

…

diagmask.cu

…

diagmask.cuh

…

dmmv.cu

…

dmmv.cuh

…

fattn-common.cuh

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn-tile-f16.cu

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn-tile-f16.cuh

…

fattn-tile-f32.cu

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn-tile-f32.cuh

…

fattn-vec-f16.cuh

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn-vec-f32.cuh

Fix FlashAttention debug test, FP32 assert (#7684)

2024-06-01 23:26:10 +02:00

fattn-wmma-f16.cuh

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn.cu

CUDA: fix Pascal FA, deq. KV to FP16 for batch > 8 (#7681)

2024-06-01 15:47:04 +02:00

fattn.cuh

…

getrows.cu

…

getrows.cuh

…

im2col.cu

…

im2col.cuh

…

mmq.cu

CUDA: quantized KV support for FA vec (#7527)

2024-06-01 08:44:14 +02:00

mmq.cuh

…

mmvq.cu

…

mmvq.cuh

…

norm.cu

…

norm.cuh

…

pad.cu

…

pad.cuh

…

pool2d.cu

…

pool2d.cuh

…

quantize.cu

…

quantize.cuh

…

rope.cu

…

rope.cuh

…

scale.cu

…

scale.cuh

…

softmax.cu

…

softmax.cuh

…

sumrows.cu

…

sumrows.cuh

…

tsembd.cu

…

tsembd.cuh

…

unary.cu

…

unary.cuh

…

upscale.cu

…

upscale.cuh

…

vecdotq.cuh

CUDA: quantized KV support for FA vec (#7527)

2024-06-01 08:44:14 +02:00