llamacpp/llama.cpp/ggml/src/ggml-cuda/ Source Tree

../
acc.cu
acc.cuh
add-id.cu
add-id.cuh
arange.cu
arange.cuh
argmax.cu
argmax.cuh
argsort.cu
argsort.cuh
binbcast.cu bin_bcast_cuda
binbcast.cuh
clamp.cu
clamp.cuh
common.cuh ggml_backend_cuda_context ggml_cuda_device_info ggml_cuda_graph ggml_cuda_mm_fusion_args_device ggml_cuda_mm_fusion_args_host ggml_cuda_pool ggml_cuda_pool_alloc ggml_cuda_unroll ggml_graph_node_properties ggml_tensor_extra_gpu
concat.cu
concat.cuh
conv-transpose-1d.cu
conv-transpose-1d.cuh
conv2d-dw.cu conv_params cwhn_layout kernel_bounds whcn_layout
conv2d-dw.cuh
conv2d-transpose.cu
conv2d-transpose.cuh
conv2d.cu conv_params kernel_bounds whcn_layout
conv2d.cuh
convert.cu
convert.cuh
count-equal.cu
count-equal.cuh
cp-async.cuh
cpy-utils.cuh
cpy.cu
cpy.cuh
cross-entropy-loss.cu
cross-entropy-loss.cuh
dequantize.cuh
diagmask.cu
diagmask.cuh
fattn-common.cuh
fattn-mma-f16.cuh fattn_mma_f16_config
fattn-tile.cu
fattn-tile.cuh
fattn-vec.cuh
fattn-wmma-f16.cu
fattn-wmma-f16.cuh
fattn.cu best_fattn_kernel
fattn.cuh
getrows.cu
getrows.cuh
ggml-cuda.cu ggml_backend_cuda_buffer_context ggml_backend_cuda_buffer_type_context ggml_backend_cuda_device_context ggml_backend_cuda_reg_context ggml_backend_cuda_split_buffer_context ggml_backend_cuda_split_buffer_type_context ggml_cuda_pool_leg
gla.cu
gla.cuh
im2col.cu
im2col.cuh
mean.cu
mean.cuh
mma.cuh tile
mmf.cu
mmf.cuh mmf_ids_data
mmid.cu mm_ids_helper_store
mmid.cuh
mmq.cu
mmq.cuh block_q8_1_mmq mmq_args mmq_q8_1_ds_layout tile_x_sizes
mmvf.cu
mmvf.cuh
mmvq.cu mmvq_parameter_table_id
mmvq.cuh
norm.cu
norm.cuh
opt-step-adamw.cu
opt-step-adamw.cuh
opt-step-sgd.cu
opt-step-sgd.cuh
out-prod.cu
out-prod.cuh
pad.cu
pad.cuh
pad_reflect_1d.cu
pad_reflect_1d.cuh
pool2d.cu
pool2d.cuh
quantize.cu
quantize.cuh
reduce_rows.cuh
roll.cu
roll.cuh
rope.cu mrope_sections rope_corr_dims
rope.cuh
scale.cu
scale.cuh
set-rows.cu
set-rows.cuh
set.cu
set.cuh
softcap.cu
softcap.cuh
softmax.cu soft_max_params
softmax.cuh
ssm-conv.cu
ssm-conv.cuh
ssm-scan.cu
ssm-scan.cuh
sum.cu
sum.cuh
sumrows.cu
sumrows.cuh
[+] template-instances/
topk-moe.cu
topk-moe.cuh
tsembd.cu
tsembd.cuh
unary.cu
unary.cuh
upscale.cu
upscale.cuh
vecdotq.cuh
[+] vendors/
wkv.cu
wkv.cuh

Browse the source code of llamacpp/llama.cpp/ggml/src/ggml-cuda/ online