bitorch_engine.functions.cuda.extension.get_ext

bitorch_engine.functions.cuda.extension.get_ext(path: Path)[source]

Generate CUDA extension for specified path.

Parameters:

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

Returns:

CUDA extension for specified path.

Return type:

Extension