hanzo-rocm 0.5.2

Rust bindings for AMD ROCm libraries
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

// src/rocblas/mod.rs

// Private modules
pub mod error;
pub mod handle;
pub mod level1;
pub mod level2;
pub mod level3;
pub mod types;
pub mod utils;
pub(crate) mod macros;
// We need to make this public for the rest of the crate
// but don't necessarily want to expose it to users
#[allow(warnings)]
pub(crate) mod bindings;

// Public re-export of FFI for internal use
mod async_ops;
pub mod ffi;

// Re-export the main components for the public API
pub use error::{Error, Result};
pub use handle::Handle;
pub use level1::{
    amax,
    amax_batched,
    amax_strided_batched,
    amin,
    amin_batched,
    amin_strided_batched,
    asum,
    // batched variants
    asum_batched,
    // strided batched variants
    asum_strided_batched,
    axpy,
    axpy_batched,
    axpy_strided_batched,
    copy,
    copy_batched,
    copy_strided_batched,
    dot,
    dot_batched,
    dot_strided_batched,
    dotc,
    dotc_batched,
    dotc_strided_batched,
    dotu,
    dotu_batched,
    dotu_strided_batched,
    nrm2,
    nrm2_batched,
    nrm2_strided_batched,
    rot,
    rot_batched,
    rot_strided_batched,
    rotg,
    rotg_batched,
    rotg_strided_batched,
    rotm,
    rotm_batched,
    rotm_strided_batched,
    rotmg,
    rotmg_batched,
    rotmg_strided_batched,
    scal,
    scal_batched,
    scal_strided_batched,
    swap,
    swap_batched,
    swap_strided_batched,
};
pub use level2::{
    gbmv,
    // batched variants
    gbmv_batched,
    // strided batched variants
    gbmv_strided_batched,
    gemv,
    gemv_batched,
    gemv_strided_batched,
    hbmv,
    hbmv_batched,
    hbmv_strided_batched,
};
pub use level3::{gemm, gemm_batched, gemm_strided_batched};
pub use types::{
    rocblas_bfloat16, rocblas_datatype, rocblas_diagonal, rocblas_double_complex, rocblas_fill,
    rocblas_float_complex, rocblas_half, rocblas_operation, rocblas_side,
};
pub use utils::{
    AtomicsMode, GemmAlgo, GemmFlags, LayerMode, MathMode, PerformanceMetric, PointerMode,
    get_atomics_mode, get_math_mode, get_performance_metric, get_pointer_mode, set_atomics_mode,
    set_math_mode, set_performance_metric, set_pointer_mode,
};

/// Create a RocBLAS handle
pub fn create_handle() -> Result<Handle> {
    Handle::new()
}

/// Initialize RocBLAS
///
/// Note: In most cases, explicit initialization is not required
/// as handle creation will initialize the library
pub fn init() -> Result<()> {
    // Creating and immediately dropping a handle
    // will initialize rocBLAS and free resources
    let _ = create_handle()?;
    Ok(())
}