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
Matthew Douglas's avatar
Apply clang-format rules (#1678)  
Matthew Douglas 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