bitorch_engine.layers.qlinear.nbit.cutlass.extension.get_ext

bitorch_engine.layers.qlinear.nbit.cutlass.extension.get_ext(path: Path)[source]

Return CUDA extension for Q Linear Cutlass.

Parameters:

path (Path) – Path to the directory containing the extension files.

Returns:

CUDA extension for Q Linear Cutlass.

Return type:

Extension