tblis-ffi 0.2.6

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

//! FFI function declarations for non-dynamic-loading.
//!
//! This file is generated automatically.

use super::*;

unsafe extern "C" {
    pub fn tblis_init_scalar_s(s: *mut tblis_scalar, value: f32);
    pub fn tblis_init_scalar_d(s: *mut tblis_scalar, value: f64);
    pub fn tblis_init_scalar_c(s: *mut tblis_scalar, value: scomplex);
    pub fn tblis_init_scalar_z(s: *mut tblis_scalar, value: dcomplex);
    pub fn tblis_init_tensor_scaled_s(
        t: *mut tblis_tensor,
        scalar: f32,
        ndim: c_int,
        len: *mut len_type,
        data: *mut f32,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_scaled_d(
        t: *mut tblis_tensor,
        scalar: f64,
        ndim: c_int,
        len: *mut len_type,
        data: *mut f64,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_scaled_c(
        t: *mut tblis_tensor,
        scalar: scomplex,
        ndim: c_int,
        len: *mut len_type,
        data: *mut scomplex,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_scaled_z(
        t: *mut tblis_tensor,
        scalar: dcomplex,
        ndim: c_int,
        len: *mut len_type,
        data: *mut dcomplex,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_s(
        t: *mut tblis_tensor,
        ndim: c_int,
        len: *mut len_type,
        data: *mut f32,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_d(
        t: *mut tblis_tensor,
        ndim: c_int,
        len: *mut len_type,
        data: *mut f64,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_c(
        t: *mut tblis_tensor,
        ndim: c_int,
        len: *mut len_type,
        data: *mut scomplex,
        stride: *mut stride_type,
    );
    pub fn tblis_init_tensor_z(
        t: *mut tblis_tensor,
        ndim: c_int,
        len: *mut len_type,
        data: *mut dcomplex,
        stride: *mut stride_type,
    );
    pub static tblis_single: *const tblis_comm;
    pub fn tblis_get_num_threads() -> c_uint;
    pub fn tblis_set_num_threads(num_threads: c_uint);
    pub fn tblis_tensor_add(
        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 fn tblis_tensor_dot(
        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 fn tblis_tensor_reduce(
        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 fn tblis_tensor_scale(
        comm: *const tblis_comm,
        cntx: *const tblis_config,
        A: *mut tblis_tensor,
        idx_A: *const label_type,
    );
    pub fn tblis_tensor_set(
        comm: *const tblis_comm,
        cntx: *const tblis_config,
        alpha: *const tblis_scalar,
        A: *mut tblis_tensor,
        idx_A: *const label_type,
    );
    pub fn tblis_tensor_shift(
        comm: *const tblis_comm,
        cntx: *const tblis_config,
        alpha: *const tblis_scalar,
        A: *mut tblis_tensor,
        idx_A: *const label_type,
    );
    pub fn tblis_tensor_mult(
        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,
    );
}