Trait grpc_vision_svc::image_captioning::utils::DeviceUtils

pub trait DeviceUtils {
    // Required methods
    fn cuda_is_available(&self) -> bool;
    fn metal_is_available(&self) -> bool;
}

A trait for device utilities. Primarily used for DI & mocking in tests.

Required Methods

fn cuda_is_available(&self) -> bool

fn metal_is_available(&self) -> bool

Implementors

impl DeviceUtils for DefaultDeviceUtils