EngineX-Ascend/enginex-ascend-910-llama.cpp
10
0
Fork 0
Code Issues Pull Requests Actions 4 Projects Releases Wiki Activity
Files
716301d1b03c31875ec3b24526c48c8b1bd0fd8c
enginex-ascend-910-llama.cpp/ggml/src/ggml-cuda
History
R0CKSTAR 716301d1b0 musa: enable fp16 mma (all) and cublas on qy2 (#13842) 
* musa: enable fp16 mma (all) and cublas on qy2

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

* Update ggml/src/ggml-cuda/ggml-cuda.cu

Co-authored-by: Johannes Gäßler <johannesg@5d6.de>

* Address review comments

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

* Address review comments

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

* musa: disable MUL_MAT_ID (q2_k × f32) due to precision issues

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

---------

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
Co-authored-by: Johannes Gäßler <johannesg@5d6.de>
2025-06-26 12:11:59 +08:00
..
template-instances
CUDA: FA support for Deepseek (Ampere or newer) (#13306)
2025-05-09 13:34:58 +02:00
vendors
CUDA/HIP: Share the same unified memory allocation logic. (#12934)
2025-04-15 11:20:38 +02:00
acc.cu
llama/ggml: add LLM training support (#10544)
2025-05-12 14:44:49 +02:00
acc.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
arange.cu
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
arange.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
argmax.cu
cuda : optimize argmax (#10441)
2024-11-21 18:18:50 +01:00
argmax.cuh
ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)
2024-10-03 21:17:26 +03:00
argsort.cu
ggml : reduce hash table reset cost (#8698)
2024-07-27 04:41:55 +02:00
argsort.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
binbcast.cu
Support pure float16 add/sub/mul/div operations in the CUDA (and CPU) backend (ggml/1121)
2025-03-03 18:18:11 +02:00
binbcast.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-20 21:15:05 +03:00
clamp.cu
cuda: unary ops as float + de-duplicate (ggml/1130)
2025-03-03 18:18:11 +02:00
clamp.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
CMakeLists.txt
CUDA: FA support for Deepseek (Ampere or newer) (#13306)
2025-05-09 13:34:58 +02:00
common.cuh
musa: enable fp16 mma (all) and cublas on qy2 (#13842)
2025-06-26 12:11:59 +08:00
concat.cu
musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)
2025-03-30 10:59:38 +02:00
concat.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
conv2d-dw.cu
CUDA: add conv_2d_dw (#14265)
2025-06-20 09:50:24 +08:00
conv2d-dw.cuh
CUDA: add conv_2d_dw (#14265)
2025-06-20 09:50:24 +08:00
conv2d-transpose.cu
CUDA: add conv_2d_transpose (#14287)
2025-06-20 22:48:24 +08:00
conv2d-transpose.cuh
CUDA: add conv_2d_transpose (#14287)
2025-06-20 22:48:24 +08:00
conv-transpose-1d.cu
musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)
2025-03-30 10:59:38 +02:00
conv-transpose-1d.cuh
feat: cuda implementation for ggml_conv_transpose_1d (ggml/854)
2024-07-08 12:23:00 +03:00
convert.cu
CUDA: fix non-cont. inputs for batched mat mul (#13155)
2025-04-29 16:00:27 +02:00
convert.cuh
CUDA: fix non-cont. inputs for batched mat mul (#13155)
2025-04-29 16:00:27 +02:00
count-equal.cu
ggml: fix zero division in ‘dne’ calculation in CUDA COUNT_EQUAL operator when ‘ne’ is small (#10213)
2024-11-09 08:35:46 +01:00
count-equal.cuh
ggml/ex: calculate accuracy in graph, adapt MNIST (ggml/980)
2024-10-03 21:17:26 +03:00
cp-async.cuh
CUDA: FA support for Deepseek (Ampere or newer) (#13306)
2025-05-09 13:34:58 +02:00
cpy.cu
musa: Upgrade MUSA SDK version to rc4.0.1 and use mudnn::Unary::IDENTITY op to accelerate D2D memory copy (#13647)
2025-05-21 09:58:49 +08:00
cpy.cuh
ggml: Re-enable CUDA graphs in presence of CONT and DUP nodes (#12970)
2025-04-17 15:19:42 +02:00
cross-entropy-loss.cu
MUSA: support ARM64 and enable dp4a .etc (#11843)
2025-02-21 09:46:23 +02:00
cross-entropy-loss.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-20 21:15:05 +03:00
dequantize.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
diagmask.cu
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
diagmask.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
fattn-common.cuh
CUDA: fix FA tg at long context for CC >= 8.9 (#13852)
2025-05-28 13:33:37 +02:00
fattn-mma-f16.cuh
CUDA: fix FTZ in FA for Gemma 3 (#13991)
2025-06-04 08:57:05 +02:00
fattn-tile-f16.cu
CUDA: FA support for Deepseek (Ampere or newer) (#13306)
2025-05-09 13:34:58 +02:00
fattn-tile-f16.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
fattn-tile-f32.cu
CUDA: FA support for Deepseek (Ampere or newer) (#13306)
2025-05-09 13:34:58 +02:00
fattn-tile-f32.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
fattn-vec-f16.cuh
CUDA: fix race condition in FA vector kernels (#13742)
2025-05-24 11:46:19 +02:00
fattn-vec-f32.cuh
CUDA: fix race condition in FA vector kernels (#13742)
2025-05-24 11:46:19 +02:00
fattn-wmma-f16.cu
musa: enable fp16 mma (all) and cublas on qy2 (#13842)
2025-06-26 12:11:59 +08:00
fattn-wmma-f16.cuh
CUDA: use mma PTX instructions for FlashAttention (#11583)
2025-02-02 19:31:09 +01:00
fattn.cu
CUDA: faster Deepseek FA, add Turing support (#13435)
2025-05-14 16:08:20 +02:00
fattn.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
getrows.cu
CUDA: fix crash on large batch size for MoE models (#13384)
2025-05-09 12:14:04 +02:00
getrows.cuh
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (#13199)
2025-04-30 23:12:59 +02:00
ggml-cuda.cu
musa: enable fp16 mma (all) and cublas on qy2 (#13842)
2025-06-26 12:11:59 +08:00
gla.cu
llama: add support for QRWKV6 model architecture (#11001)
2025-01-10 09:58:08 +08:00
gla.cuh
llama: add support for QRWKV6 model architecture (#11001)
2025-01-10 09:58:08 +08:00
im2col.cu
CUDA: fix 1D im2col, add tests (ggml/993)
2024-10-23 16:50:02 +03:00
im2col.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
mean.cu
CUDA: add mean operation (#14313)
2025-06-22 12:39:54 +08:00
mean.cuh
CUDA: add mean operation (#14313)
2025-06-22 12:39:54 +08:00
mma.cuh
musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)
2025-03-30 10:59:38 +02:00
mmq.cu
CUDA: fix crash on large batch size for quant. MoE (#13537)
2025-05-14 16:41:02 +02:00
mmq.cuh
cuda : remove nrows_x in mul_mat_q_process_tile (#13325)
2025-05-07 09:48:23 +02:00
mmv.cu
CUDA/HIP: optimize mmv paths taken for HIP devices (#14324)
2025-06-24 01:12:56 +02:00
mmv.cuh
CUDA: mul_mat_v support for batch sizes > 1 (#14262)
2025-06-23 13:11:31 +02:00
mmvq.cu
CUDA: fix crash with partial offloading of MoE (#13439)
2025-05-11 16:09:33 +02:00
mmvq.cuh
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (#13014)
2025-04-22 21:27:40 +02:00
norm.cu
llama: Add support for RWKV v7 architecture (#12412)
2025-03-18 07:27:50 +08:00
norm.cuh
llama: Add support for RWKV v7 architecture (#12412)
2025-03-18 07:27:50 +08:00
opt-step-adamw.cu
ggml: new optimization interface (ggml/988)
2024-11-17 08:30:29 +02:00
opt-step-adamw.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-20 21:15:05 +03:00
out-prod.cu
CPU/CUDA: fix (GQA) mul mat back, add CUDA support (#11380)
2025-01-24 12:38:31 +01:00
out-prod.cuh
ggml/examples: add backend support for numerical optimization (ggml/949)
2024-09-20 21:15:05 +03:00
pad.cu
musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)
2025-03-30 10:59:38 +02:00
pad.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
pool2d.cu
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
pool2d.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
quantize.cu
CUDA: fix crash on large batch size for quant. MoE (#13537)
2025-05-14 16:41:02 +02:00
quantize.cuh
CUDA: batched+noncont MMQ, refactor bs>1 MoE code (#13199)
2025-04-30 23:12:59 +02:00
rope.cu
CUDA: backwards pass for misc. ops, add tests (#11257)
2025-01-16 16:43:38 +01:00
rope.cuh
RoPE: fix back, CUDA support for back + noncont. (#11240)
2025-01-15 12:51:37 +01:00
scale.cu
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
scale.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
softmax.cu
HIP: fix flash_attn_stream_k_fixup warning (#11604)
2025-02-02 23:48:29 +01:00
softmax.cuh
CUDA: backwards pass for misc. ops, add tests (#11257)
2025-01-16 16:43:38 +01:00
ssm-conv.cu
fix MUSA compiler warning (#12704)
2025-04-03 09:32:55 +02:00
ssm-conv.cuh
ggml : faster ssm scan (#10558)
2025-03-31 18:05:13 +02:00
ssm-scan.cu
CUDA/HIP: fix ssm_scan on devices where warp size is not 32 (#14196)
2025-06-15 17:30:13 +02:00
ssm-scan.cuh
ggml : faster ssm scan (#10558)
2025-03-31 18:05:13 +02:00
sum.cu
llama/ggml: add LLM training support (#10544)
2025-05-12 14:44:49 +02:00
sum.cuh
tests: add gradient tests for all backends (ggml/932)
2024-09-08 11:05:55 +03:00
sumrows.cu
CUDA: add mean operation (#14313)
2025-06-22 12:39:54 +08:00
sumrows.cuh
CUDA: add mean operation (#14313)
2025-06-22 12:39:54 +08:00
tsembd.cu
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
tsembd.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
unary.cu
ggml : add ggml_gelu_erf() CUDA kernel (#13719)
2025-05-24 13:06:47 +02:00
unary.cuh
ggml : add ggml_gelu_erf() CUDA kernel (#13719)
2025-05-24 13:06:47 +02:00
upscale.cu
musa: fix all warnings, re-enable -DLLAMA_FATAL_WARNINGS=ON in ci and update doc (#12611)
2025-03-30 10:59:38 +02:00
upscale.cuh
llama : reorganize source code + improve CMake (#8006)
2024-06-26 18:33:02 +03:00
vecdotq.cuh
CUDA: noncont MMVQ + batched bs1 MUL_MAT_ID (#13014)
2025-04-22 21:27:40 +02:00
wkv.cu
llama: Add support for RWKV v7 architecture (#12412)
2025-03-18 07:27:50 +08:00
wkv.cuh
llama: Add support for RWKV v7 architecture (#12412)
2025-03-18 07:27:50 +08:00