From 3b46b033b594fb92fd81c2b48abd2093b50f787c Mon Sep 17 00:00:00 2001
From: Zhang Shuo <52872288+fuyou4546@users.noreply.github.com>
Date: Tue, 11 Aug 2026 02:42:15 +0000
Subject: [PATCH] feat(nvidia): add DeviceGetAttribute to runtime

---
 src/native/cuda/nvidia/runtime_.h | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/native/cuda/nvidia/runtime_.h b/src/native/cuda/nvidia/runtime_.h
index 2fb53cd..99e87f6 100644
--- a/src/native/cuda/nvidia/runtime_.h
+++ b/src/native/cuda/nvidia/runtime_.h
@@ -38,6 +38,19 @@ struct Runtime<Device::Type::kNvidia>
 
   static constexpr auto GetDeviceCount = cudaGetDeviceCount;
 
+  static constexpr auto DeviceGetAttribute = cudaDeviceGetAttribute;
+
+  static constexpr auto kDevAttrWarpSize = cudaDevAttrWarpSize;
+
+  static constexpr auto kDevAttrComputeCapabilityMajor =
+      cudaDevAttrComputeCapabilityMajor;
+
+  static constexpr auto kDevAttrComputeCapabilityMinor =
+      cudaDevAttrComputeCapabilityMinor;
+
+  static constexpr auto kDevAttrMaxSharedMemoryPerBlockOptin =
+      cudaDevAttrMaxSharedMemoryPerBlockOptin;
+
   static constexpr auto DeviceSynchronize = cudaDeviceSynchronize;
 
   static constexpr auto Malloc = [](auto&&... args) {