Skip to content

Commit 57fbe06

Browse files
committed
Fix RDNA4 perf
1 parent 29f39f8 commit 57fbe06

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
@@ -1281,7 +1281,7 @@ static void ggml_cuda_op_mul_mat_cublas(
12811281

12821282
CUBLAS_CHECK(cublasSetStream(ctx.cublas_handle(id), stream));
12831283

1284-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1284+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
12851285
const float alpha = 1.0f;
12861286
const float beta = 0.0f;
12871287
CUBLAS_CHECK(
@@ -1850,7 +1850,7 @@ static void ggml_cuda_mul_mat_batched_cublas(ggml_backend_cuda_context & ctx, co
18501850

18511851
int id = ggml_cuda_get_device();
18521852
const int cc = ggml_cuda_info().devices[id].cc;
1853-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1853+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
18541854
cu_compute_type = CUBLAS_COMPUTE_32F;
18551855
alpha = &alpha_f32;
18561856
beta = &beta_f32;

0 commit comments

Comments
 (0)