common.h 413 Bytes
Newer Older
Max Ryabinin's avatar
Max Ryabinin committed
1
2
3
4
5
6
7
#include <BinSearch.h>

#ifndef common
#define common

using namespace BinSearch;

8
9
#define BLOCK_SIZE 16384

Max Ryabinin's avatar
Max Ryabinin committed
10
11
12
13
14
15
struct quantize_block_args {
    BinAlgo<Scalar, float, Direct2> *bin_searcher;
    float *code;
    float *A;
    float *absmax;
    unsigned char *out;
16
17
18
19
    long long block_end;
    long long block_idx;
    long long threadidx;
		long long blocksize;
Max Ryabinin's avatar
Max Ryabinin committed
20
21
22
23
24
};


void *quantize_block(void *arguments);

Max Ryabinin's avatar
Max Ryabinin committed
25
#endif