Skip to main content

setup_cuda_env

Function setup_cuda_env 

Source
pub fn setup_cuda_env() -> Result<CudaTestEnv>
Expand description

Asserts CUDA is available, opens the first device, and resolves its compute capability (overridable via the TEENYC_CAPABILITY env var, e.g. sm_90) — the standard setup for teeny-cuda’s own device-dependent tests.