bitorch_engine.layers.qlinear.binary.cutlass.extension.get_ext

bitorch_engine.layers.qlinear.binary.cutlass.extension.get_ext(path: Path)

Get the CUDA extension for binary linear cutlass.

Parameters:

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

Returns:

The CUDA extension for binary linear cutlass.

Return type:

Extension