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

Fix nvrtcGetErrorString.

Browse source
This commit is contained in:
Seunghoon Lee 2024-03-21 22:06:42 +09:00
parent 87d2e3e163
commit ff1bc6d9b6
No known key found for this signature in database
GPG key ID: 436E38F4E70BD152
2 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
zluda_rtc/src/lib.rs
View file

@ -19,19 +19,22 @@ fn to_nvrtc(status: hiprtc_sys::hiprtcResult) -> nvrtcResult {
match status {
hiprtc_sys::hiprtcResult::HIPRTC_SUCCESS => nvrtcResult::NVRTC_SUCCESS,
hiprtc_sys::hiprtcResult::HIPRTC_ERROR_INVALID_PROGRAM => nvrtcResult::NVRTC_ERROR_INVALID_PROGRAM,
hiprtc_sys::hiprtcResult::HIPRTC_ERROR_COMPILATION => nvrtcResult::NVRTC_ERROR_COMPILATION,
err => panic!("[ZLUDA] HIPRTC failed: {}", err.0),
}
}
fn get_error_string(result: nvrtcResult) -> *const ::std::os::raw::c_char {
let error_string =
match result {
nvrtcResult::NVRTC_ERROR_INTERNAL_ERROR => String::from("NVRTC_ERROR_INTERNAL_ERROR"),
_ => result.0.to_string(),
};
println!("[ZLUDA] HIPRTC failed: {}", error_string);
let cstr = std::ffi::CString::new("").unwrap();
cstr.as_ptr()
fn to_hiprtc(status: nvrtcResult) -> hiprtc_sys::hiprtcResult {
match status {
nvrtcResult::NVRTC_SUCCESS => hiprtc_sys::hiprtcResult::HIPRTC_SUCCESS,
nvrtcResult::NVRTC_ERROR_INVALID_PROGRAM => hiprtc_sys::hiprtcResult::HIPRTC_ERROR_INVALID_PROGRAM,
nvrtcResult::NVRTC_ERROR_COMPILATION => hiprtc_sys::hiprtcResult::HIPRTC_ERROR_COMPILATION,
err => panic!("[ZLUDA] HIPRTC failed: {}", err.0),
}
}
unsafe fn get_error_string(result: nvrtcResult) -> *const ::std::os::raw::c_char {
hiprtcGetErrorString(to_hiprtc(result))
}
unsafe fn create_program(

2
zluda_rtc/src/nvrtc.rs
View file

@ -43,7 +43,7 @@ pub struct nvrtcResult(pub ::std::os::raw::c_int);
#[doc = " \\ingroup error\n \\brief nvrtcGetErrorString is a helper function that returns a string\n describing the given nvrtcResult code, e.g., NVRTC_SUCCESS to\n \\c \"NVRTC_SUCCESS\".\n For unrecognized enumeration values, it returns\n \\c \"NVRTC_ERROR unknown\".\n\n \\param [in] result CUDA Runtime Compilation API result code.\n \\return Message string for the given #nvrtcResult code."]
#[no_mangle]
pub extern "system" fn nvrtcGetErrorString(result: nvrtcResult) -> *const ::std::os::raw::c_char {
pub unsafe extern "system" fn nvrtcGetErrorString(result: nvrtcResult) -> *const ::std::os::raw::c_char {
crate::get_error_string(result)
}