bitorch_engine.layers.qlinear.binary.cuda.extension.get_ext

bitorch_engine.layers.qlinear.binary.cuda.extension.get_ext(path: Path)[source]

Get the CUDA extension for binary linear operations.

Parameters:

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

Returns:

The CUDA extension module.

Return type:

Any