Skip to main content

select_visible_gpu_archs

Function select_visible_gpu_archs 

Source
pub fn select_visible_gpu_archs(
    enumerated_archs: &[String],
    rocr_visible_devices: Option<&str>,
    hip_visible_devices: Option<&str>,
    cuda_visible_devices: Option<&str>,
) -> Vec<String>
Expand description

Apply the first configured visibility mask to enumerated discrete GPUs.