ggml-blas.h 586 Bytes
Newer Older
xuxzh1's avatar
init  
xuxzh1 committed
1
2
3
4
5
6
7
8
9
10
11
#pragma once

#include "ggml.h"
#include "ggml-backend.h"


#ifdef  __cplusplus
extern "C" {
#endif

// backend API
xuxzh1's avatar
update  
xuxzh1 committed
12
GGML_BACKEND_API ggml_backend_t ggml_backend_blas_init(void);
xuxzh1's avatar
init  
xuxzh1 committed
13

xuxzh1's avatar
update  
xuxzh1 committed
14
GGML_BACKEND_API bool ggml_backend_is_blas(ggml_backend_t backend);
xuxzh1's avatar
init  
xuxzh1 committed
15
16
17

// number of threads used for conversion to float
// for openblas and blis, this will also set the number of threads used for blas operations
xuxzh1's avatar
update  
xuxzh1 committed
18
19
20
GGML_BACKEND_API void ggml_backend_blas_set_n_threads(ggml_backend_t backend_blas, int n_threads);

GGML_BACKEND_API ggml_backend_reg_t ggml_backend_blas_reg(void);
xuxzh1's avatar
init  
xuxzh1 committed
21
22
23
24
25


#ifdef  __cplusplus
}
#endif