bitsandbytes-rocm/csrc/cpu_ops.h

11 lines
335 B
C
Raw Normal View History

2022-07-01 14:16:10 +00:00
#ifndef BITSANDBYTES_CPU_OPS_H
#define BITSANDBYTES_CPU_OPS_H
#include <iostream>
#include <stdio.h>
2022-07-01 14:16:10 +00:00
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);
2022-07-01 14:16:10 +00:00
#endif