diff --git a/zluda_bindgen/src/main.rs b/zluda_bindgen/src/main.rs index 05582c0..a1f8023 100644 --- a/zluda_bindgen/src/main.rs +++ b/zluda_bindgen/src/main.rs @@ -24,22 +24,22 @@ static KNOWN_CUDA_VERSIONS: &[&'static str] = &[ fn main() { let crate_root = PathBuf::from_str(env!("CARGO_MANIFEST_DIR")).unwrap(); - // generate_hip_runtime( - // &crate_root, - // &["..", "ext", "hip_runtime-sys", "src", "lib.rs"], - // ); + generate_hip_runtime( + &crate_root, + &["..", "ext", "hip_runtime-sys", "src", "lib.rs"], + ); generate_rocblas( &crate_root, &["..", "ext", "rocblas-sys", "src", "lib.rs"], ); - // let cuda_functions = generate_cuda(&crate_root); - // generate_process_address_table(&crate_root, cuda_functions); - // generate_ml(&crate_root); - // generate_cublas(&crate_root); - // generate_cublaslt(&crate_root); - // generate_cufft(&crate_root); - // generate_cusparse(&crate_root); - // generate_cudnn(&crate_root); + let cuda_functions = generate_cuda(&crate_root); + generate_process_address_table(&crate_root, cuda_functions); + generate_ml(&crate_root); + generate_cublas(&crate_root); + generate_cublaslt(&crate_root); + generate_cufft(&crate_root); + generate_cusparse(&crate_root); + generate_cudnn(&crate_root); } fn generate_process_address_table(crate_root: &PathBuf, mut cuda_fns: Vec) {