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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
|
/*
* phylokernelfma.cpp
*
* Created on: Sept 25, 2016
* Author: minh
*/
#include "vectorclass/vectormath_exp.h"
#include "vectorclass/vectorclass.h"
#include "phylokernel.h"
//#include "phylokernelsafe.h"
//#include "phylokernelmixture.h"
//#include "phylokernelmixrate.h"
//#include "phylokernelsitemodel.h"
#include "phylokernelnew.h"
#include "phylokernelnonrev.h"
#define KERNEL_FIX_STATES
#include "phylokernelnew.h"
#include "phylokernelnonrev.h"
#if !defined(__AVX2__) && !defined(__FMA__)
#error "You must compile this file with AVX2 or FMA enabled!"
#endif
void PhyloTree::setDotProductFMA() {
#ifdef BOOT_VAL_FLOAT
dotProduct = &PhyloTree::dotProductSIMD<float, Vec8f>;
#else
dotProduct = &PhyloTree::dotProductSIMD<double, Vec4d>;
#endif
dotProductDouble = &PhyloTree::dotProductSIMD<double, Vec4d>;
}
void PhyloTree::setLikelihoodKernelFMA() {
vector_size = 4;
bool site_model = model_factory && model_factory->model->isSiteSpecificModel();
// setParsimonyKernelAVX();
if (site_model && ((model_factory && !model_factory->model->isReversible()) || params->kernel_nonrev))
outError("Site-specific model is not yet supported for nonreversible models");
computeLikelihoodDervMixlenPointer = NULL;
if (site_model && safe_numeric) {
// safe site-specific model
switch (aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, SAFE_LH, 4, true, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, SAFE_LH, 4, true, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, SAFE_LH, 4, true, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 4, true, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, SAFE_LH, 20, true, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, SAFE_LH, 20, true, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, SAFE_LH, 20, true, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 20, true, true>;
break;
default:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchGenericSIMD <Vec4d, SAFE_LH, true, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervGenericSIMD <Vec4d, SAFE_LH, true, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodGenericSIMD <Vec4d, SAFE_LH, true, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferGenericSIMD<Vec4d, true, true>;
break;
}
return;
}
if (site_model) {
switch (aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, NORM_LH, 4, true, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, NORM_LH, 4, true, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, NORM_LH, 4, true, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 4, true, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, NORM_LH, 20, true, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, NORM_LH, 20, true, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, NORM_LH, 20, true, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 20, true, true>;
break;
default:
ASSERT(0);
break;
}
return;
}
if ((model_factory && !model_factory->model->isReversible()) || params->kernel_nonrev) {
// if nonreversible model
if (safe_numeric)
switch (aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchSIMD <Vec4d, SAFE_LH, 4, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervSIMD <Vec4d, SAFE_LH, 4, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodSIMD<Vec4d, SAFE_LH, 4, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchSIMD <Vec4d, SAFE_LH, 20, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervSIMD <Vec4d, SAFE_LH, 20, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodSIMD<Vec4d, SAFE_LH, 20, true>;
break;
default:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchGenericSIMD <Vec4d, SAFE_LH, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervGenericSIMD <Vec4d, SAFE_LH, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodGenericSIMD<Vec4d, SAFE_LH, true>;
break;
} else {
switch (aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchSIMD <Vec4d, NORM_LH, 4, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervSIMD <Vec4d, NORM_LH, 4, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodSIMD<Vec4d, NORM_LH, 4, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchSIMD <Vec4d, NORM_LH, 20, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervSIMD <Vec4d, NORM_LH, 20, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodSIMD<Vec4d, NORM_LH, 20, true>;
break;
default:
computeLikelihoodBranchPointer = &PhyloTree::computeNonrevLikelihoodBranchGenericSIMD <Vec4d, NORM_LH, true>;
computeLikelihoodDervPointer = &PhyloTree::computeNonrevLikelihoodDervGenericSIMD <Vec4d, NORM_LH, true>;
computePartialLikelihoodPointer = &PhyloTree::computeNonrevPartialLikelihoodGenericSIMD<Vec4d, NORM_LH, true>;
break;
}
}
computeLikelihoodFromBufferPointer = NULL;
return;
}
if (safe_numeric) {
switch(aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, SAFE_LH, 4, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, SAFE_LH, 4, true>;
computeLikelihoodDervMixlenPointer = &PhyloTree::computeLikelihoodDervMixlenSIMD<Vec4d, SAFE_LH, 4, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, SAFE_LH, 4, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 4, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, SAFE_LH, 20, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, SAFE_LH, 20, true>;
computeLikelihoodDervMixlenPointer = &PhyloTree::computeLikelihoodDervMixlenSIMD<Vec4d, SAFE_LH, 20, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, SAFE_LH, 20, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 20, true>;
break;
default:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchGenericSIMD <Vec4d, SAFE_LH, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervGenericSIMD <Vec4d, SAFE_LH, true>;
computeLikelihoodDervMixlenPointer = &PhyloTree::computeLikelihoodDervMixlenGenericSIMD<Vec4d, SAFE_LH, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodGenericSIMD <Vec4d, SAFE_LH, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferGenericSIMD<Vec4d, true>;
break;
}
return;
}
switch(aln->num_states) {
case 4:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, NORM_LH, 4, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, NORM_LH, 4, true>;
computeLikelihoodDervMixlenPointer = &PhyloTree::computeLikelihoodDervMixlenSIMD<Vec4d, NORM_LH, 4, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, NORM_LH, 4, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 4, true>;
break;
case 20:
computeLikelihoodBranchPointer = &PhyloTree::computeLikelihoodBranchSIMD <Vec4d, NORM_LH, 20, true>;
computeLikelihoodDervPointer = &PhyloTree::computeLikelihoodDervSIMD <Vec4d, NORM_LH, 20, true>;
computeLikelihoodDervMixlenPointer = &PhyloTree::computeLikelihoodDervMixlenSIMD<Vec4d, NORM_LH, 20, true>;
computePartialLikelihoodPointer = &PhyloTree::computePartialLikelihoodSIMD <Vec4d, NORM_LH, 20, true>;
computeLikelihoodFromBufferPointer = &PhyloTree::computeLikelihoodFromBufferSIMD<Vec4d, 20, true>;
break;
default:
ASSERT(0);
break;
}
}
|
