From 843170d404a51df849f3eb781a625e1c252d18fb Mon Sep 17 00:00:00 2001 From: m0_53063391 <1710065346@qq.com> Date: Wed, 8 Jul 2026 22:58:21 +0800 Subject: [PATCH] fix: cuda version >= 13.0 has bug --- 02-mixed-precision-gemm/mixed_precision_gemm.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/02-mixed-precision-gemm/mixed_precision_gemm.cu b/02-mixed-precision-gemm/mixed_precision_gemm.cu index 97e7bc4..2dee5d8 100644 --- a/02-mixed-precision-gemm/mixed_precision_gemm.cu +++ b/02-mixed-precision-gemm/mixed_precision_gemm.cu @@ -4,7 +4,7 @@ #include #include -#if (defined(__CUDA_ARCH__) && (__CUDA_ARCH__ >= 890) && ((__CUDACC_VER_MAJOR__ >= 12) && (__CUDACC_VER_MINOR__ >= 4))) +#if (defined(__CUDA_ARCH__) && (__CUDA_ARCH__ >= 890) && (__CUDACC_VER_MAJOR__ * 100 + __CUDACC_VER_MINOR__ >= 1204)) #define CUTE_ARCH_MMA_SM89_ENABLED #endif @@ -126,7 +126,7 @@ struct SM90_16x8x32_F32E4M3E5M2F32_TN { : "=f"(d0), "=f"(d1), "=f"(d2), "=f"(d3) : "r"(a0), "r"(a1), "r"(a2), "r"(a3), "r"(b0), "r"(b1), "f"(c0), "f"(c1), "f"(c2), "f"(c3)); #else - CUTE_INVALID_CONTROL_PATH("Attempting to use SM90_16x8x32_F32E4M3E5M2F32_TN without CUTE_ARCH_MMA_SM89_ENABLED"); + CUTE_INVALID_CONTROL_PATH("Attempting to use SM90_16x8x32_F32E4M3E5M2F32_TN without CUTE_ARCH_MMA_SM89_ENABLED\n"); #endif } };