Apppublic

Aluode/PerceptionLabPortable

sourceHugging Faceupdated 9mo agoView on Hugging Face

0likes

rootpython_embedLibsite-packagessklearnutils_cython_blas.pxd

_cython_blas.pxd42 linesDownload Raw Back to utils

1from cython cimport floating
2
3
4cpdef enum BLAS_Order:
5    RowMajor  # C contiguous
6    ColMajor  # Fortran contiguous
7
8
9cpdef enum BLAS_Trans:
10    NoTrans = 110  # correspond to 'n'
11    Trans = 116    # correspond to 't'
12
13
14# BLAS Level 1 ################################################################
15cdef floating _dot(int, const floating*, int, const floating*, int) noexcept nogil
16
17cdef floating _asum(int, const floating*, int) noexcept nogil
18
19cdef void _axpy(int, floating, const floating*, int, floating*, int) noexcept nogil
20
21cdef floating _nrm2(int, const floating*, int) noexcept nogil
22
23cdef void _copy(int, const floating*, int, const floating*, int) noexcept nogil
24
25cdef void _scal(int, floating, const floating*, int) noexcept nogil
26
27cdef void _rotg(floating*, floating*, floating*, floating*) noexcept nogil
28
29cdef void _rot(int, floating*, int, floating*, int, floating, floating) noexcept nogil
30
31# BLAS Level 2 ################################################################
32cdef void _gemv(BLAS_Order, BLAS_Trans, int, int, floating, const floating*, int,
33                const floating*, int, floating, floating*, int) noexcept nogil
34
35cdef void _ger(BLAS_Order, int, int, floating, const floating*, int, const floating*,
36               int, floating*, int) noexcept nogil
37
38# BLASLevel 3 ################################################################
39cdef void _gemm(BLAS_Order, BLAS_Trans, BLAS_Trans, int, int, int, floating,
40                const floating*, int, const floating*, int, floating, floating*,
41                int) noexcept nogil
42