bitorch_engine.layers.qconv.binary.cutlass.extension.get_ext

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

Obtains the CUDA extension for Cutlass-based binary convolution.

Parameters:

path (Path) – The path to the directory containing the necessary source files for the Cutlass-based binary convolution operation.

Returns:

The CUDA extension for the Cutlass-based binary convolution.

Return type:

Extension