tblis-ffi 0.2.4

FFI bindings of TBLIS
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
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

//! Library struct definition for dynamic-loading.
//!
//! This file is generated automatically.

use super::*;

pub struct DyLoadLib {
    pub __libraries: Vec<libloading::Library>,
    pub __libraries_path: Vec<String>,
    pub tblis_init_scalar_s: Option<unsafe extern "C" fn(s: *mut tblis_scalar, value: f32)>,
    pub tblis_init_scalar_d: Option<unsafe extern "C" fn(s: *mut tblis_scalar, value: f64)>,
    pub tblis_init_scalar_c: Option<unsafe extern "C" fn(s: *mut tblis_scalar, value: scomplex)>,
    pub tblis_init_scalar_z: Option<unsafe extern "C" fn(s: *mut tblis_scalar, value: dcomplex)>,
    pub tblis_init_tensor_scaled_s: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            scalar: f32,
            ndim: c_int,
            len: *mut len_type,
            data: *mut f32,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_scaled_d: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            scalar: f64,
            ndim: c_int,
            len: *mut len_type,
            data: *mut f64,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_scaled_c: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            scalar: scomplex,
            ndim: c_int,
            len: *mut len_type,
            data: *mut scomplex,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_scaled_z: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            scalar: dcomplex,
            ndim: c_int,
            len: *mut len_type,
            data: *mut dcomplex,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_s: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            ndim: c_int,
            len: *mut len_type,
            data: *mut f32,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_d: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            ndim: c_int,
            len: *mut len_type,
            data: *mut f64,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_c: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            ndim: c_int,
            len: *mut len_type,
            data: *mut scomplex,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_init_tensor_z: Option<
        unsafe extern "C" fn(
            t: *mut tblis_tensor,
            ndim: c_int,
            len: *mut len_type,
            data: *mut dcomplex,
            stride: *mut stride_type,
        ),
    >,
    pub tblis_get_num_threads: Option<unsafe extern "C" fn() -> c_uint>,
    pub tblis_set_num_threads: Option<unsafe extern "C" fn(num_threads: c_uint)>,
    pub tblis_tensor_add: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            A: *const tblis_tensor,
            idx_A: *const label_type,
            B: *mut tblis_tensor,
            idx_B: *const label_type,
        ),
    >,
    pub tblis_tensor_dot: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            A: *const tblis_tensor,
            idx_A: *const label_type,
            B: *const tblis_tensor,
            idx_B: *const label_type,
            result: *mut tblis_scalar,
        ),
    >,
    pub tblis_tensor_reduce: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            op: reduce_t,
            A: *const tblis_tensor,
            idx_A: *const label_type,
            result: *mut tblis_scalar,
            idx: *mut len_type,
        ),
    >,
    pub tblis_tensor_scale: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            A: *mut tblis_tensor,
            idx_A: *const label_type,
        ),
    >,
    pub tblis_tensor_set: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            alpha: *const tblis_scalar,
            A: *mut tblis_tensor,
            idx_A: *const label_type,
        ),
    >,
    pub tblis_tensor_shift: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            alpha: *const tblis_scalar,
            A: *mut tblis_tensor,
            idx_A: *const label_type,
        ),
    >,
    pub tblis_tensor_mult: Option<
        unsafe extern "C" fn(
            comm: *const tblis_comm,
            cntx: *const tblis_config,
            A: *const tblis_tensor,
            idx_A: *const label_type,
            B: *const tblis_tensor,
            idx_B: *const label_type,
            C: *mut tblis_tensor,
            idx_C: *const label_type,
        ),
    >,
}