From 7127592069c2fe01b041e174ba4345ef9b279671 Mon Sep 17 00:00:00 2001 From: dePaul Miller Date: Mon, 5 Jan 2026 10:50:44 -0800 Subject: [PATCH] Replace CUDA driver API with runtime API (#2928) Co-authored-by: dePaul Miller <23461061+depaulmillz@users.noreply.github.com> --- include/cute/atom/copy_traits_sm90_im2col.hpp | 4 ++-- include/cute/atom/copy_traits_sm90_tma.hpp | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/include/cute/atom/copy_traits_sm90_im2col.hpp b/include/cute/atom/copy_traits_sm90_im2col.hpp index d2cc0dce..06a19dbf 100644 --- a/include/cute/atom/copy_traits_sm90_im2col.hpp +++ b/include/cute/atom/copy_traits_sm90_im2col.hpp @@ -472,8 +472,8 @@ make_im2col_tma_copy_desc( tma_oob_fill); int driver_version = 0; - CUresult driver_version_result = cuDriverGetVersion(&driver_version); - assert(driver_version_result == CUDA_SUCCESS); + cudaError_t driver_version_err = cudaDriverGetVersion(&driver_version); + assert(driver_version_err == cudaSuccess); if (driver_version <= 13010) { if (cute::bits_to_bytes( cute::cosize(tensor_cwhdn.layout()) * diff --git a/include/cute/atom/copy_traits_sm90_tma.hpp b/include/cute/atom/copy_traits_sm90_tma.hpp index f69ec7b7..90dedd0b 100644 --- a/include/cute/atom/copy_traits_sm90_tma.hpp +++ b/include/cute/atom/copy_traits_sm90_tma.hpp @@ -1053,8 +1053,8 @@ make_tma_copy_desc(Tensor const& gtensor, // The origin tma_oobFill); int driver_version = 0; - CUresult driver_version_result = cuDriverGetVersion(&driver_version); - assert(driver_version_result == CUDA_SUCCESS); + cudaError_t driver_version_err = cudaDriverGetVersion(&driver_version); + assert(driver_version_err == cudaSuccess); if (driver_version <= 13010) { if (cute::bits_to_bytes( cute::cosize(gtensor.layout()) *