Skip to content

GitLab

Unverified Commit b23ce920 authored by Tyler Michael Smith's avatar Tyler Michael Smith Committed by GitHub
Browse files

[Bugfix] Fix CUDA version check for mma warning suppression (#5642)

parent 2bd231a7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
csrc/quantization/marlin/sparse/common/mma.h
View file @ b23ce920
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
#pragma once #pragma once
#include "base.h" #include "base.h"
#include <cudaTypedefs.h>
namespace marlin_24 { namespace marlin_24 {
...@@ -26,7 +27,7 @@ namespace marlin_24 { ...@@ -26,7 +27,7 @@ namespace marlin_24 {
// | Advisory: Modifier ‘.sp::ordered_metadata’ should be used on instruction // | Advisory: Modifier ‘.sp::ordered_metadata’ should be used on instruction
// | ‘mma’ instead of modifier ‘.sp’ as it is expected to have substantially // | ‘mma’ instead of modifier ‘.sp’ as it is expected to have substantially
// | reduced performance on some future architectures // | reduced performance on some future architectures
#if defined CUDA_VERSION && CUDA_VERSION >= 12500 #if defined CUDA_VERSION && CUDA_VERSION >= 12050
#define MMA_SP_INST \ #define MMA_SP_INST \
"mma.sp::ordered_metadata.sync.aligned.m16n8k32.row.col.f32.f16.f16.f32 " "mma.sp::ordered_metadata.sync.aligned.m16n8k32.row.col.f32.f16.f16.f32 "
#else #else
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment