mistralrs_quant/gptq/
mod.rs

1#[cfg(feature = "cuda")]
2mod ffi;
3#[cfg(not(feature = "cuda"))]
4mod gptq_cpu;
5#[cfg(feature = "cuda")]
6mod gptq_cuda;
7#[cfg(feature = "cuda")]
8mod marlin_backend;
9#[cfg(feature = "cuda")]
10mod marlin_ffi;
11
12#[cfg(not(feature = "cuda"))]
13pub use gptq_cpu::{gptq_linear, GptqLayer};
14#[cfg(feature = "cuda")]
15pub use gptq_cuda::{gptq_linear, GptqLayer};