-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathtaco_kernels.c
61 lines (57 loc) · 903 Bytes
/
taco_kernels.c
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
double taco_sum_mul2() {
load_ssA();
load_ssB();
#include "taco/sum_mul2.c"
}
double taco_sum_add2() {
load_ssA();
load_ssB();
#include "taco/sum_add2.c"
}
double taco_sum_mul2_csr() {
load_ssA();
load_dsB();
#include "taco/sum_mul2_csr.c"
}
double taco_inner2ss() {
load_ssA();
load_ssB();
#include "taco/inner2ss.c"
}
double taco_wcoj() {
load_ssR();
load_ssT();
#include "taco/wcoj.c"
}
double taco_mttkrp() {
load_dsA();
load_dsB();
load_sssC();
//printf("TODO\n");
#include "taco/mttkrp.c"
return 0;
}
double taco_sum_mul2_inner() {
load_ssA();
load_ssB();
load_dsA();
load_dsB();
#include "taco/sum_mul2_inner.c"
}
double taco_sum_mul2_inner_ss() {
load_ssA();
load_ssB();
#include "taco/sum_mul2_inner_ss.c"
}
double taco_spmv() {
load_ssA();
load_dV();
#include "taco/spmv.c"
}
double taco_filter_spmv() {
//load_sV();
//load_dsA();
//#include "taco/spmv.c"
return 0.0;
}
/* here end */