blob: 4a2475273f8537e58d7837704987694324c7a61a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#![crate_type = "bin"]
use std::ffi::c_void;
use std::mem;
use std::env;
use std::path::PathBuf;
use std::ffi::CString;
extern "system" {
fn LoadLibraryA(lpFileName: *const i8) -> *mut c_void;
fn GetProcAddress(hModule: *mut c_void, lpProcName: *const u8) -> *mut c_void;
}
fn main() {
let current_exe = env::current_exe().unwrap();
let mut dll = PathBuf::from(current_exe.parent().unwrap());
dll.push("do_cuinit.dll");
let dll_cstring = CString::new(dll.to_str().unwrap()).unwrap();
let nvcuda = unsafe { LoadLibraryA(dll_cstring.as_ptr()) };
let cuInit = unsafe { GetProcAddress(nvcuda, b"do_cuinit\0".as_ptr()) };
let cuInit = unsafe { mem::transmute::<_, unsafe extern "system" fn(u32) -> u32>(cuInit) };
unsafe { cuInit(0) };
}
|