rocm-rs 0.4.2

Rust bindings for AMD ROCm libraries
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

// src/rocfft/mod.rs

//! Bindings for rocfft
//! Auto-generated - do not modify
#[allow(warnings)]
pub mod bindings;
pub mod cache;
pub mod description;
pub mod error;
pub mod execution;
pub mod ffi;
pub mod field;
pub mod plan;

// Add the new utility modules
pub mod examples;
pub mod utils;

// Re-export all bindings
pub use bindings::*;

/// Initialize rocFFT library
pub fn setup() -> error::Result<()> {
    unsafe { error::check_error(bindings::rocfft_setup()) }
}

/// Cleanup rocFFT library
pub fn cleanup() -> error::Result<()> {
    unsafe { error::check_error(bindings::rocfft_cleanup()) }
}

/// Get the rocFFT version string
pub fn get_version() -> error::Result<String> {
    let mut buffer = vec![0u8; 100];
    unsafe {
        error::check_error(bindings::rocfft_get_version_string(
            buffer.as_mut_ptr() as *mut i8,
            buffer.len(),
        ))?;

        // Find the null terminator
        let len = buffer.iter().position(|&c| c == 0).unwrap_or(buffer.len());
        buffer.truncate(len);

        Ok(String::from_utf8_lossy(&buffer).to_string())
    }
}