Server crashing on first prompt

#3
by belx-dev - opened

I've succesfully used assistant models with both the dense and MOE models on RTX 4090 (win) with meaningful speed up. However, the E4B drafter keeps crashing on me. The server starts successfully but then immediately crashes after receiving first prompt.

I'm quite new to this, so I'm probably missing something obvious.

FIXED:
Not sure, whether the problem was in my local branch, but Claude has fixed for me after several trial and errors. The problem was in ggml/src/ggml-cuda/fattn.cu, ggml/src/ggml-cuda/fattn-mma-f16.cuh, ggml/src/ggml-cuda/template-instances/fattn-mma-f16-instance-ncols1_*-ncols2_2.cu. When flash attention MMA is used with head dimension D=512 and a GQA ratio of exactly 2 (i.e. n_head / n_head_kv == 2), the server crashes unconditionally with fatal error.

I got crash also.
This is log:

image

Sign up or log in to comment