rocm-rs 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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168

// src/rocblas/mod.rs

// Private modules
pub mod error;
pub mod handle;
pub mod level1;
pub mod level2;
pub mod level3;
pub mod safe;
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,
    gerc,
    gerc_batched,
    gerc_strided_batched,
    ger,
    ger_batched,
    ger_strided_batched,
    geru,
    geru_batched,
    geru_strided_batched,
    hbmv,
    hbmv_batched,
    hbmv_strided_batched,
    hemm,
    hemv,
    hemv_batched,
    hemv_strided_batched,
    herk,
    spr,
    spr2,
    spr2_batched,
    spr2_strided_batched,
    spr_batched,
    spr_strided_batched,
    syr,
    syr2,
    syr2_batched,
    syr2_strided_batched,
    syr_batched,
    syr_strided_batched,
    symv,
    trmv,
    trsv,
    tpmv,
    tpsv,
    tbmv,
    tbsv,
    spmv,
    sbmv,
    hpmv,
    hpr,
    hpr2,
};
pub use level3::{
    gemm,
    gemm_batched,
    gemm_ex,
    gemm_strided_batched,
    geam,
    hemm_batched,
    hemm_strided_batched,
    herk_batched,
    herk_strided_batched,
    herkx,
    herkx_batched,
    herkx_strided_batched,
    symm,
    syr2k,
    syrk,
    trmm,
    trsm,
};
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(())
}