Skip to main content

select_backend

Function select_backend 

Source
pub fn select_backend(size: usize, gpu_available: bool) -> BackendCategory
Expand description

Select optimal backend for ML operation based on data size

§Arguments

  • size - Number of elements to process
  • gpu_available - Whether GPU is available

§Returns

Recommended backend category

§Example

use aprender::compute::{select_backend, BackendCategory};

let category = select_backend(50_000, false);
assert_eq!(category, BackendCategory::SimdParallel);