bitorch_engine.layers.qlinear.nbit.cuda.extension.get_ext

bitorch_engine.layers.qlinear.nbit.cuda.extension.get_ext(path: Path)

Get the CUDA extension for quantized linear operations.

Parameters:

path (Path) – The path to the CUDA extension directory.

Returns:

The CUDA extension module.

Return type:

Any