From b23ce9203235488e080434108d3504d54b24e867 Mon Sep 17 00:00:00 2001 From: Tyler Michael Smith Date: Tue, 18 Jun 2024 19:48:49 -0400 Subject: [PATCH] [Bugfix] Fix CUDA version check for mma warning suppression (#5642) --- csrc/quantization/marlin/sparse/common/mma.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/csrc/quantization/marlin/sparse/common/mma.h b/csrc/quantization/marlin/sparse/common/mma.h index 8a6c65338f277..b26505f771c8b 100644 --- a/csrc/quantization/marlin/sparse/common/mma.h +++ b/csrc/quantization/marlin/sparse/common/mma.h @@ -17,6 +17,7 @@ #pragma once #include "base.h" +#include namespace marlin_24 { @@ -26,7 +27,7 @@ namespace marlin_24 { // | Advisory: Modifier ‘.sp::ordered_metadata’ should be used on instruction // | ‘mma’ instead of modifier ‘.sp’ as it is expected to have substantially // | reduced performance on some future architectures -#if defined CUDA_VERSION && CUDA_VERSION >= 12500 +#if defined CUDA_VERSION && CUDA_VERSION >= 12050 #define MMA_SP_INST \ "mma.sp::ordered_metadata.sync.aligned.m16n8k32.row.col.f32.f16.f16.f32 " #else