bitorch_engine.utils.cuda_extension

Functions

gcc_version

Determines the version of GCC (GNU Compiler Collection) installed on the system.

get_cuda_arch

get_cuda_extension

get_kwargs

Generates keyword arguments for compilation based on the GCC version and CUDA architecture.