MCPcopy Create free account
hub / github.com/IST-DASLab/MoE-Quant / gptq_loop

Function gptq_loop

src/gptq_loop.py:193–225  ·  view source on GitHub ↗

Quantize weight tensor with GPTQ algorithm weight: (C, R), transposed weight tensor to quantize, modified in-place and returned hessian_inv: (C, C), inverse of Hessian matrix scale: (C, R), transposed scale tensor for quantization qzero: (C, R), transposed zero-point tensor for

(
    weight: torch.Tensor,
    hessian_inv: torch.Tensor,
    scale: torch.Tensor,
    qzero: torch.Tensor,
    maxq: torch.Tensor,
    dtype: torch.dtype,
    gptq_block_size: int = 128,
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

gptq_loop_graphFunction · 0.85

Tested by

no test coverage detected