Skip to content

Commit a90c9f4

Browse files
committed
Fix RDNA4 perf
1 parent 3e574df commit a90c9f4

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
@@ -1285,7 +1285,7 @@ static void ggml_cuda_op_mul_mat_cublas(
12851285

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

1288-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1288+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
12891289
const float alpha = 1.0f;
12901290
const float beta = 0.0f;
12911291
CUBLAS_CHECK(
@@ -1902,7 +1902,7 @@ static void ggml_cuda_mul_mat_batched_cublas_impl(ggml_backend_cuda_context & ct
19021902

19031903
int id = ggml_cuda_get_device();
19041904
const int cc = ggml_cuda_info().devices[id].cc;
1905-
if (GGML_CUDA_CC_IS_CDNA(cc) || GGML_CUDA_CC_IS_RDNA4(cc)) {
1905+
if (GGML_CUDA_CC_IS_CDNA(cc)) {
19061906
cu_compute_type = CUBLAS_COMPUTE_32F;
19071907
alpha = &alpha_f32;
19081908
beta = &beta_f32;

0 commit comments

Comments
 (0)