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 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) {