Revert "Implement vectorization (UNTESTED)"
This reverts commit 04ecc147fb.
This commit is contained in:
Jacob
230
cpu-opt/main.cpp
230
cpu-opt/main.cpp
@@ -1,6 +1,5 @@
|
|||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
#include <immintrin.h>
|
#include <immintrin.h>
|
||||||
#include "aligned_allocator.cpp"
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
@@ -163,10 +162,10 @@ Coxeter E8() {
|
|||||||
struct RelTable {
|
struct RelTable {
|
||||||
std::vector<int> coset_poss;
|
std::vector<int> coset_poss;
|
||||||
std::vector<Cos> init_cosets;
|
std::vector<Cos> init_cosets;
|
||||||
std::vector<Cos,aligned_allocator<Cos,ALIGN_SIZE>> start_cosets;
|
std::vector<Cos> start_cosets;
|
||||||
std::vector<Cos,aligned_allocator<Cos,ALIGN_SIZE>> end_cosets;
|
std::vector<Cos> end_cosets;
|
||||||
std::vector<Ind,aligned_allocator<Cos,ALIGN_SIZE>> start_inds;
|
std::vector<Ind> start_inds;
|
||||||
std::vector<Ind,aligned_allocator<Cos,ALIGN_SIZE>> end_inds;
|
std::vector<Ind> end_inds;
|
||||||
int num_rows;
|
int num_rows;
|
||||||
Gen gen[2];
|
Gen gen[2];
|
||||||
Ind end_ind;
|
Ind end_ind;
|
||||||
@@ -271,6 +270,9 @@ int add_coset(const Coxeter &cox,
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//#define __AVX2__
|
||||||
|
|
||||||
#ifndef __AVX2__
|
#ifndef __AVX2__
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -352,224 +354,6 @@ void learn(const Coxeter &cox, CosetTable &cosets,
|
|||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
const auto mask = _mm256_set1_epi32(1);
|
|
||||||
inline void compute_lookups(Ind* inds, Cos* coss, Gen* gens, Cos* base, __m256i step, Cos* tar) {
|
|
||||||
_mm256_store_si256(
|
|
||||||
(__m256i*)tar,
|
|
||||||
_mm256_i32gather_epi32(
|
|
||||||
(int*)base,
|
|
||||||
_mm256_add_epi32(
|
|
||||||
_mm256_mullo_epi32(
|
|
||||||
_mm256_load_si256((__m256i*)coss),
|
|
||||||
step
|
|
||||||
),
|
|
||||||
_mm256_i32gather_epi32(
|
|
||||||
gens,
|
|
||||||
_mm256_and_si256(
|
|
||||||
_mm256_load_si256((__m256i*)inds),
|
|
||||||
mask
|
|
||||||
),
|
|
||||||
4
|
|
||||||
)
|
|
||||||
),
|
|
||||||
4
|
|
||||||
)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* learn until it can't
|
|
||||||
*/
|
|
||||||
void learn(const Coxeter &cox, CosetTable &cosets,
|
|
||||||
std::vector<RelTable> &reltables) {
|
|
||||||
|
|
||||||
const int nrels = cox.nrels;
|
|
||||||
const int ngens = cox.ngens;
|
|
||||||
const auto step = _mm256_set1_epi32(ngens);
|
|
||||||
|
|
||||||
while (true) {
|
|
||||||
bool complete = true;
|
|
||||||
|
|
||||||
alignas(32) Gen gens[2];
|
|
||||||
alignas(32) Cos lookups[8];
|
|
||||||
alignas(32) Cos init_cosets[8];
|
|
||||||
alignas(32) Ind start_inds[8];
|
|
||||||
alignas(32) Ind end_inds[8];
|
|
||||||
alignas(32) Cos start_cosets[8];
|
|
||||||
alignas(32) Cos end_cosets[8];
|
|
||||||
#pragma omp parallel for schedule(static, 1) reduction(&:complete) private(gens, lookups, init_cosets, start_inds, end_inds, start_cosets, end_cosets)
|
|
||||||
for (unsigned int r = 0; r < nrels; ++r) {
|
|
||||||
auto &table = reltables[r];
|
|
||||||
gens[0] = table.gen[0];
|
|
||||||
gens[1] = table.gen[1];
|
|
||||||
|
|
||||||
unsigned int c;
|
|
||||||
bool redo_cval;
|
|
||||||
for (c = 0; c < ((table.num_rows>>3)<<3); c+=8) {
|
|
||||||
redo_cval = false;
|
|
||||||
for (int c_ = 0; c_ < 8; c_++) {
|
|
||||||
init_cosets[c_] = table.init_cosets[c+c_];
|
|
||||||
start_inds[c_] = table.start_inds[c+c_];
|
|
||||||
end_inds[c_] = table.end_inds[c+c_];
|
|
||||||
start_cosets[c_] = table.start_cosets[c+c_];
|
|
||||||
end_cosets[c_] = table.end_cosets[c+c_];
|
|
||||||
}
|
|
||||||
|
|
||||||
bool startdone = false;
|
|
||||||
bool need_reload = false;
|
|
||||||
int reload_idx;
|
|
||||||
int idx, lookup;
|
|
||||||
while (!startdone) {
|
|
||||||
startdone = true;
|
|
||||||
if (need_reload and idx < table.num_rows) {
|
|
||||||
init_cosets[reload_idx] = table.init_cosets[idx];
|
|
||||||
start_inds[reload_idx] = table.start_inds[idx];
|
|
||||||
end_inds[reload_idx] = table.end_inds[idx];
|
|
||||||
start_cosets[reload_idx] = table.start_cosets[idx];
|
|
||||||
end_cosets[reload_idx] = table.end_cosets[idx];
|
|
||||||
}
|
|
||||||
compute_lookups(start_inds, start_cosets, gens, &(cosets[0]), step, lookups);
|
|
||||||
for (int c_ = 0; c_ < 8; c_++) {
|
|
||||||
lookup = lookups[c_];
|
|
||||||
if (start_inds[c_] < end_inds[c_] and lookup >= 0) {
|
|
||||||
start_inds[c_]++;
|
|
||||||
start_cosets[c_] = lookups[c_];
|
|
||||||
startdone = false;
|
|
||||||
if (lookup > init_cosets[c_]) {
|
|
||||||
int idx = table.coset_poss[lookup];
|
|
||||||
if (idx >= 0) {
|
|
||||||
table.rem_row(idx);
|
|
||||||
if ( (idx>>3)<<3 == c ) {
|
|
||||||
need_reload = true;
|
|
||||||
reload_idx = (idx & 8);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool enddone = false;
|
|
||||||
while (!enddone) {
|
|
||||||
enddone = true;
|
|
||||||
if (need_reload and idx < table.num_rows) {
|
|
||||||
init_cosets[reload_idx] = table.init_cosets[idx];
|
|
||||||
start_inds[reload_idx] = table.start_inds[idx];
|
|
||||||
end_inds[reload_idx] = table.end_inds[idx];
|
|
||||||
start_cosets[reload_idx] = table.start_cosets[idx];
|
|
||||||
end_cosets[reload_idx] = table.end_cosets[idx];
|
|
||||||
}
|
|
||||||
compute_lookups(end_inds, end_cosets, gens, &(cosets[0]), step, lookups);
|
|
||||||
for (int c_ = 0; c_ < 8; c_++) {
|
|
||||||
lookup = lookups[c_];
|
|
||||||
if (start_inds[c_] < end_inds[c_] and lookup >= 0) {
|
|
||||||
end_inds[c_]--;
|
|
||||||
end_cosets[c_] = lookups[c_];
|
|
||||||
enddone = false;
|
|
||||||
if (lookup > init_cosets[c_]) {
|
|
||||||
int idx = table.coset_poss[lookup];
|
|
||||||
if (idx >= 0) {
|
|
||||||
table.rem_row(idx);
|
|
||||||
if ( (idx>>3)<<3 == c ) {
|
|
||||||
need_reload = true;
|
|
||||||
reload_idx = (idx & 8);
|
|
||||||
redo_cval = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (int c_ = 0; c_ < 8 and c+c_ < table.num_rows; c_++) {
|
|
||||||
table.start_inds[c+c_] = start_inds[c_];
|
|
||||||
table.end_inds[c+c_] = end_inds[c_];
|
|
||||||
table.start_cosets[c+c_] = start_cosets[c_];
|
|
||||||
table.end_cosets[c+c_] = end_cosets[c_];
|
|
||||||
}
|
|
||||||
|
|
||||||
for (int c_ = 8; c_ >= 0; c_--) {
|
|
||||||
if (c+c_ < table.num_rows)
|
|
||||||
continue;
|
|
||||||
Ind s_i = start_inds[c_];
|
|
||||||
if (start_inds[s_i] == end_inds[c_]) {
|
|
||||||
complete = false;
|
|
||||||
|
|
||||||
const Gen gen = gens[s_i&1];
|
|
||||||
Cos s_c = start_cosets[c_];
|
|
||||||
Cos e_c = end_cosets[c_];
|
|
||||||
cosets[s_c*ngens + gen] = e_c;
|
|
||||||
cosets[e_c*ngens + gen] = s_c;
|
|
||||||
|
|
||||||
table.rem_row(c+c_);
|
|
||||||
redo_cval = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (redo_cval)
|
|
||||||
c-=8;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (; c < table.num_rows; c++) {
|
|
||||||
auto s_i = table.start_inds[c];
|
|
||||||
auto e_i = table.end_inds[c];
|
|
||||||
auto s_c = table.start_cosets[c];
|
|
||||||
auto e_c = table.end_cosets[c];
|
|
||||||
auto i_c = table.init_cosets[c];
|
|
||||||
|
|
||||||
while (s_i < e_i) {
|
|
||||||
const Cos lookup = cosets[s_c*ngens + gens[s_i&1]];
|
|
||||||
if (lookup < 0) break;
|
|
||||||
|
|
||||||
s_i++;
|
|
||||||
s_c = lookup;
|
|
||||||
|
|
||||||
if (s_c > i_c) {
|
|
||||||
int idx = table.coset_poss[s_c];
|
|
||||||
if (idx >= 0)
|
|
||||||
table.rem_row(idx);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
table.start_inds[c] = s_i;
|
|
||||||
table.start_cosets[c] = s_c;
|
|
||||||
|
|
||||||
while (s_i < e_i) {
|
|
||||||
const Cos lookup = cosets[e_c*ngens + gens[e_i&1]];
|
|
||||||
if (lookup < 0) break;
|
|
||||||
|
|
||||||
e_i--;
|
|
||||||
e_c = lookup;
|
|
||||||
|
|
||||||
if (e_c > i_c) {
|
|
||||||
int idx = table.coset_poss[e_c];
|
|
||||||
if (idx >= 0)
|
|
||||||
table.rem_row(idx);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
table.end_inds[c] = e_i;
|
|
||||||
table.end_cosets[c] = e_c;
|
|
||||||
|
|
||||||
if (s_i == e_i) {
|
|
||||||
complete = false;
|
|
||||||
|
|
||||||
const Gen gen = gens[s_i&1];
|
|
||||||
cosets[s_c*ngens + gen] = e_c;
|
|
||||||
cosets[e_c*ngens + gen] = s_c;
|
|
||||||
|
|
||||||
table.rem_row(c);
|
|
||||||
c--;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (complete) break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
CosetTable solve_tc(const Coxeter &cox, const Gens &subgens) {
|
CosetTable solve_tc(const Coxeter &cox, const Gens &subgens) {
|
||||||
|
|||||||
Reference in New Issue
Block a user