c8h4/ZLUDA
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Implement nvmlDeviceGetCudaComputeCapability.

Browse source
This commit is contained in:
Seunghoon Lee 2024-04-21 00:31:57 +09:00
parent 12c2da499f
commit cf262c6889
No known key found for this signature in database
GPG key ID: 436E38F4E70BD152
2 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
zluda_ml/src/common.rs
View file

@ -67,6 +67,8 @@ pub(crate) fn error_string(result: nvmlReturn_t) -> *const ::std::os::raw::c_cha
// https://docs.nvidia.com/deploy/cuda-compatibility/index.html#minor-version-compatibility
const VERSION: &'static [u8] = b"511.09.01"; // minimum required by Arnold
const NVML_VERSION: &'static [u8] = b"11.511.09.01"; // // minimum required by Arnold
const NVML_COMPUTE_CAPABILITY_MAJOR: i32 = 8;
const NVML_COMPUTE_CAPABILITY_MINOR: i32 = 8;
impl From<Result<(), nvmlReturn_t>> for nvmlReturn_t {
fn from(result: Result<(), nvmlReturn_t>) -> Self {
@ -158,3 +160,13 @@ pub(crate) unsafe fn parse_pci_address(address: *const i8) -> Result<(i32, i32,
i32::from_str_radix(bus, 16).map_err(|_| nvmlReturn_t::NVML_ERROR_INVALID_ARGUMENT)?;
Ok((bus, device, function))
}
pub(crate) unsafe fn device_get_cuda_compute_capability(
_device: *mut crate::nvml::nvmlDevice_st,
major: *mut i32,
minor: *mut i32,
) -> nvmlReturn_t {
*major = NVML_COMPUTE_CAPABILITY_MAJOR;
*minor = NVML_COMPUTE_CAPABILITY_MINOR;
nvmlReturn_t::NVML_SUCCESS
}

2
zluda_ml/src/nvml.rs
View file

@ -2868,7 +2868,7 @@ pub extern "C" fn nvmlDeviceGetCudaComputeCapability(
major: *mut ::std::os::raw::c_int,
minor: *mut ::std::os::raw::c_int,
) -> nvmlReturn_t {
crate::common::unimplemented()
unsafe { crate::common::device_get_cuda_compute_capability(device, major, minor) }
}
#[no_mangle]