OpenBLAS/interface/tobf16.c at v0.3.21 · OpenMathLib/OpenBLAS · GitHub

Name: OpenBLAS/interface/tobf16.c at v0.3.21 · OpenMathLib/OpenBLAS · GitHub
Rating: 4.7 (5260 reviews)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#include<stdio.h>
#include"common.h"
#ifdefFUNCTION_PROFILE
#include"functable.h"
#endif

#if defined(DOUBLE_PREC)
#defineFLOAT_TYPE double
#elif defined(SINGLE_PREC)
#defineFLOAT_TYPE float
#else
#endif

#ifndefCBLAS
voidNAME(blasint*N, FLOAT_TYPE*in, blasint*INC_IN, bfloat16*out, blasint*INC_OUT){
BLASLONGn=*N;
BLASLONGinc_in=*INC_IN;
BLASLONGinc_out=*INC_OUT;

PRINT_DEBUG_NAME;

if (n <= 0) return;

IDEBUG_START;
FUNCTION_PROFILE_START();

if (inc_in<0) in-= (n-1) *inc_in;
if (inc_out<0) out-= (n-1) *inc_out;

#if defined(DOUBLE_PREC)
D_TO_BF16_K(n, in, inc_in, out, inc_out);
#elif defined(SINGLE_PREC)
S_TO_BF16_K(n, in, inc_in, out, inc_out);
#else
#endif

FUNCTION_PROFILE_END(1, 2*n, 2*n);
IDEBUG_END;
}
#else
voidCNAME(blasintn, FLOAT_TYPE*in, blasintinc_in, bfloat16*out, blasintinc_out){
PRINT_DEBUG_CNAME;

if (n <= 0) return;

IDEBUG_START;
FUNCTION_PROFILE_START();

if (inc_in<0) in-= (n-1) *inc_in;
if (inc_out<0) out-= (n-1) *inc_out;

#if defined(DOUBLE_PREC)
D_TO_BF16_K(n, in, inc_in, out, inc_out);
#elif defined(SINGLE_PREC)
S_TO_BF16_K(n, in, inc_in, out, inc_out);
#endif

FUNCTION_PROFILE_END(1, 2*n, 2*n);
IDEBUG_END;
}
#endif