bitorch_engine.layers.qconv.nbit.cutlass.extension.get_ext

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

Get CUDA extension for a specified path.

Parameters:

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

Returns:

The CUDA extension object.

Return type:

Extension