mistralrs_core/cuda/
ffi.rs

1use std::ffi::c_void;
2
3#[allow(dead_code)]
4extern "C" {
5    pub(crate) fn asort_asc_f32(
6        x: *const c_void,
7        dst: *mut c_void,
8        nrows: i32,
9        ncols: i32,
10        inplace: bool,
11        stream: i64,
12    );
13    pub(crate) fn asort_asc_f16(
14        x: *const c_void,
15        dst: *mut c_void,
16        nrows: i32,
17        ncols: i32,
18        inplace: bool,
19        stream: i64,
20    );
21    pub(crate) fn asort_asc_bf16(
22        x: *const c_void,
23        dst: *const c_void,
24        nrows: i32,
25        ncols: i32,
26        inplace: bool,
27        stream: i64,
28    );
29    pub(crate) fn asort_asc_f64(
30        x: *const c_void,
31        dst: *mut c_void,
32        nrows: i32,
33        ncols: i32,
34        inplace: bool,
35        stream: i64,
36    );
37    pub(crate) fn asort_asc_u8(
38        x: *const c_void,
39        dst: *mut c_void,
40        nrows: i32,
41        ncols: i32,
42        inplace: bool,
43        stream: i64,
44    );
45    pub(crate) fn asort_asc_u32(
46        x: *const c_void,
47        dst: *mut c_void,
48        nrows: i32,
49        ncols: i32,
50        inplace: bool,
51        stream: i64,
52    );
53    pub(crate) fn asort_asc_i64(
54        x: *const c_void,
55        dst: *mut c_void,
56        nrows: i32,
57        ncols: i32,
58        inplace: bool,
59        stream: i64,
60    );
61    pub(crate) fn asort_desc_f32(
62        x: *const c_void,
63        dst: *mut c_void,
64        nrows: i32,
65        ncols: i32,
66        inplace: bool,
67        stream: i64,
68    );
69    pub(crate) fn asort_desc_f16(
70        x: *const c_void,
71        dst: *mut c_void,
72        nrows: i32,
73        ncols: i32,
74        inplace: bool,
75        stream: i64,
76    );
77    pub(crate) fn asort_desc_bf16(
78        x: *const c_void,
79        dst: *mut c_void,
80        nrows: i32,
81        ncols: i32,
82        inplace: bool,
83        stream: i64,
84    );
85    pub(crate) fn asort_desc_f64(
86        x: *const c_void,
87        dst: *mut c_void,
88        nrows: i32,
89        ncols: i32,
90        inplace: bool,
91        stream: i64,
92    );
93    pub(crate) fn asort_desc_u8(
94        x: *const c_void,
95        dst: *mut c_void,
96        nrows: i32,
97        ncols: i32,
98        inplace: bool,
99        stream: i64,
100    );
101    pub(crate) fn asort_desc_u32(
102        x: *const c_void,
103        dst: *mut c_void,
104        nrows: i32,
105        ncols: i32,
106        inplace: bool,
107        stream: i64,
108    );
109    pub(crate) fn asort_desc_i64(
110        x: *const c_void,
111        dst: *mut c_void,
112        nrows: i32,
113        ncols: i32,
114        inplace: bool,
115        stream: i64,
116    );
117}
mistralrs_core/cuda/ffi.rs

mistralrs_core/cuda/
ffi.rs
