You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
182 lines
2.9 KiB
182 lines
2.9 KiB
# Copyright ExplsionAI GmbH, released under BSD.
|
|
from cython cimport view
|
|
from libc.stdint cimport int64_t
|
|
|
|
|
|
ctypedef float[::1] float1d_t
|
|
ctypedef double[::1] double1d_t
|
|
ctypedef float[:, ::1] float2d_t
|
|
ctypedef double[:, ::1] double2d_t
|
|
ctypedef float* floats_t
|
|
ctypedef double* doubles_t
|
|
ctypedef const float[::1] const_float1d_t
|
|
ctypedef const double[::1] const_double1d_t
|
|
ctypedef const float[:, ::1] const_float2d_t
|
|
ctypedef const double[:, ::1] const_double2d_t
|
|
ctypedef const float* const_floats_t
|
|
ctypedef const double* const_doubles_t
|
|
|
|
|
|
|
|
cdef fused reals_ft:
|
|
floats_t
|
|
doubles_t
|
|
float1d_t
|
|
double1d_t
|
|
|
|
cdef fused const_reals_ft:
|
|
const_floats_t
|
|
const_doubles_t
|
|
const_float1d_t
|
|
const_double1d_t
|
|
|
|
|
|
cdef fused reals1d_ft:
|
|
float1d_t
|
|
double1d_t
|
|
|
|
cdef fused const_reals1d_ft:
|
|
const_float1d_t
|
|
const_double1d_t
|
|
|
|
|
|
cdef fused reals2d_ft:
|
|
float2d_t
|
|
double2d_t
|
|
|
|
|
|
cdef fused const_reals2d_ft:
|
|
const_float2d_t
|
|
const_double2d_t
|
|
|
|
|
|
cdef fused real_ft:
|
|
float
|
|
double
|
|
|
|
|
|
ctypedef int64_t dim_t
|
|
ctypedef int64_t inc_t
|
|
ctypedef int64_t doff_t
|
|
|
|
|
|
# Sucks to set these from magic numbers, but it's better than dragging
|
|
# the header into our header.
|
|
# We get some piece of mind from checking the values on init.
|
|
cpdef enum trans_t:
|
|
NO_TRANSPOSE = 0
|
|
TRANSPOSE = 8
|
|
CONJ_NO_TRANSPOSE = 16
|
|
CONJ_TRANSPOSE = 24
|
|
|
|
|
|
cpdef enum conj_t:
|
|
NO_CONJUGATE = 0
|
|
CONJUGATE = 16
|
|
|
|
|
|
cpdef enum side_t:
|
|
LEFT = 0
|
|
RIGHT = 1
|
|
|
|
|
|
cpdef enum uplo_t:
|
|
LOWER = 192
|
|
UPPER = 96
|
|
DENSE = 224
|
|
|
|
|
|
cpdef enum diag_t:
|
|
NONUNIT_DIAG = 0
|
|
UNIT_DIAG = 256
|
|
|
|
|
|
cdef void gemm(
|
|
trans_t transa,
|
|
trans_t transb,
|
|
dim_t m,
|
|
dim_t n,
|
|
dim_t k,
|
|
double alpha,
|
|
reals_ft a, inc_t rsa, inc_t csa,
|
|
reals_ft b, inc_t rsb, inc_t csb,
|
|
double beta,
|
|
reals_ft c, inc_t rsc, inc_t csc,
|
|
) nogil
|
|
|
|
|
|
cdef void ger(
|
|
conj_t conjx,
|
|
conj_t conjy,
|
|
dim_t m,
|
|
dim_t n,
|
|
double alpha,
|
|
reals_ft x, inc_t incx,
|
|
reals_ft y, inc_t incy,
|
|
reals_ft a, inc_t rsa, inc_t csa
|
|
) nogil
|
|
|
|
|
|
cdef void gemv(
|
|
trans_t transa,
|
|
conj_t conjx,
|
|
dim_t m,
|
|
dim_t n,
|
|
real_ft alpha,
|
|
reals_ft a, inc_t rsa, inc_t csa,
|
|
reals_ft x, inc_t incx,
|
|
real_ft beta,
|
|
reals_ft y, inc_t incy
|
|
) nogil
|
|
|
|
|
|
cdef void axpyv(
|
|
conj_t conjx,
|
|
dim_t m,
|
|
real_ft alpha,
|
|
reals_ft x, inc_t incx,
|
|
reals_ft y, inc_t incy
|
|
) nogil
|
|
|
|
|
|
cdef void scalv(
|
|
conj_t conjalpha,
|
|
dim_t m,
|
|
real_ft alpha,
|
|
reals_ft x, inc_t incx
|
|
) nogil
|
|
|
|
|
|
cdef double dotv(
|
|
conj_t conjx,
|
|
conj_t conjy,
|
|
dim_t m,
|
|
reals_ft x,
|
|
reals_ft y,
|
|
inc_t incx,
|
|
inc_t incy,
|
|
) nogil
|
|
|
|
|
|
cdef double norm_L1(
|
|
dim_t n,
|
|
reals_ft x, inc_t incx
|
|
) nogil
|
|
|
|
|
|
cdef double norm_L2(
|
|
dim_t n,
|
|
reals_ft x, inc_t incx
|
|
) nogil
|
|
|
|
|
|
cdef double norm_inf(
|
|
dim_t n,
|
|
reals_ft x, inc_t incx
|
|
) nogil
|
|
|
|
|
|
cdef void randv(
|
|
dim_t m,
|
|
reals_ft x, inc_t incx
|
|
) nogil
|