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

#ifndef common
#define common

using namespace BinSearch;

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

20
void quantize_block(const quantize_block_args& args);
Max Ryabinin's avatar
Max Ryabinin committed
21

Max Ryabinin's avatar
Max Ryabinin committed
22
#endif