common.h 428 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
void quantize_block(const quantize_block_args& args);
Max Ryabinin's avatar
Max Ryabinin committed
24

Max Ryabinin's avatar
Max Ryabinin committed
25
#endif