Skip to content

Commit 76ce720

Browse files
committed
Fix RDNA4 perf
1 parent 29f8ec1 commit 76ce720

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-cuda/ggml-cuda.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1284,7 +1284,7 @@ static void ggml_cuda_op_mul_mat_cublas(
12841284

12851285
CUBLAS_CHECK(cublasSetStream(ctx.cublas_handle(id), stream));
12861286

1287-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1287+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
12881288
const float alpha = 1.0f;
12891289
const float beta = 0.0f;
12901290
CUBLAS_CHECK(
@@ -1853,7 +1853,7 @@ static void ggml_cuda_mul_mat_batched_cublas(ggml_backend_cuda_context & ctx, co
18531853

18541854
int id = ggml_cuda_get_device();
18551855
const int cc = ggml_cuda_info().devices[id].cc;
1856-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1856+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
18571857
cu_compute_type = CUBLAS_COMPUTE_32F;
18581858
alpha = &alpha_f32;
18591859
beta = &beta_f32;

0 commit comments

Comments
 (0)