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
|
#pragma once
// Header for a test C library being wrapped by numpysane_pywrap. This library
// can compute inner and outer products
#include <stdint.h>
// Inner product supports arbitrary strides, and 3 data types
#define DECLARE_INNER_T(T) \
T inner_ ## T(const T* a, \
const T* b, \
int stride_a, \
int stride_b, \
int n);
DECLARE_INNER_T(int32_t)
DECLARE_INNER_T(int64_t)
DECLARE_INNER_T(double)
// Outer product supports arbitrary strides, and only the "double" data type
void outer(double* out,
int stride_out_incol,
int stride_out_inrow,
const double* a,
const double* b,
int stride_a,
int stride_b,
int n);
// inner and outer product together. Only contiguous data is supported. "double"
// only. non-broadcasted "scale" argument scales the output
double innerouter(double* out,
const double* a,
const double* b,
double scale,
int n);
// Assumes that indices_order[] has room for at least N values
void sorted_indices_float(// output
int* indices_order,
// input
const float* x, int N);
void sorted_indices_double(// output
int* indices_order,
// input
const double* x, int N);
|
