Skip to content

Commit 42d8205

Browse files
committed
Fix RDNA4 perf
1 parent 3c8d430 commit 42d8205

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(
@@ -1901,7 +1901,7 @@ static void ggml_cuda_mul_mat_batched_cublas_impl(ggml_backend_cuda_context & ct
19011901

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

0 commit comments

Comments
 (0)