Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cpu_ops.h 335 Bytes
Newer Older
Max Ryabinin's avatar
Add a CPU-only build option  
Max Ryabinin committed
1
2
3 
#ifndef BITSANDBYTES_CPU_OPS_H
#define BITSANDBYTES_CPU_OPS_H
Tim Dettmers's avatar
Fixed 2^31 max size issue for cpu blockwise quant.  
Tim Dettmers committed
4
5 
#include <iostream>
#include <stdio.h>
Max Ryabinin's avatar
Add a CPU-only build option  
Max Ryabinin committed
6
Tim Dettmers's avatar
Fixed 2^31 max size issue for cpu blockwise quant.  
Tim Dettmers committed
7
8 
void quantize_cpu(float *code, float *A, float *absmax, unsigned char *out, long long blocksize, long long n);
void dequantize_cpu(float *code, unsigned char *A, float *absmax, float *out, long long blocksize, long long n);
Max Ryabinin's avatar
Add a CPU-only build option  
Max Ryabinin committed
9
10 

#endif