Mirror/aicodix___code
1
0
Fork 0
mirror of https://github.com/aicodix/code.git synced 2026-04-27 14:30:36 +00:00
Code Issues Projects Releases Packages Wiki Activity

do not rely on long double

Browse source 
see this project if more precision is needed:

https://github.com/xdsopl/TwoInt/blob/main/polar.cc
This commit is contained in:
Ahmet Inan 2023-07-07 10:27:15 +02:00
commit 55ddae0c6f
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
polar_freezer.hh
View file

@ -21,7 +21,7 @@ class PolarFreezer
bits[idx/32] &= ~(1 << (idx%32));
bits[idx/32] |= (uint32_t)val << (idx%32);
}
static void freeze(uint32_t *bits, long double pe, long double th, int i, int h)
static void freeze(uint32_t *bits, double pe, double th, int i, int h)
{
if (h) {
freeze(bits, pe * (2-pe), th, i, h/2);
@ -31,7 +31,7 @@ class PolarFreezer
}
}
public:
int operator()(uint32_t *frozen_bits, int level, long double erasure_probability = 0.5L, long double freezing_threshold = 0.5L)
int operator()(uint32_t *frozen_bits, int level, double erasure_probability = 0.5, double freezing_threshold = 0.5)
{
int length = 1 << level;
freeze(frozen_bits, erasure_probability, freezing_threshold, 0, length / 2);
@ -53,7 +53,7 @@ class PolarCodeConst0
{
bits[idx/32] |= 1 << (idx%32);
}
void compute(long double pe, int i, int h)
void compute(double pe, int i, int h)
{
if (h) {
compute(pe * (2-pe), i, h/2);
@ -62,10 +62,10 @@ class PolarCodeConst0
prob[i] = pe;
}
}
long double prob[1<<MAX_M];
double prob[1<<MAX_M];
int index[1<<MAX_M];
public:
void operator()(uint32_t *frozen_bits, int level, int K, long double erasure_probability = std::exp(-1.L))
void operator()(uint32_t *frozen_bits, int level, int K, double erasure_probability = std::exp(-1.))
{
assert(level <= MAX_M);
int length = 1 << level;