Function with_linalg_runtime 

pub fn with_linalg_runtime<T, R>(
    op: &'static str,
    capability: LinalgCapabilityOp,
    cpu: impl FnOnce(&mut CpuContext) -> Result<R, Error>,
    cuda: impl FnOnce(&mut CudaContext) -> Result<R, Error>,
    rocm: impl FnOnce(&mut RocmContext) -> Result<R, Error>,
) -> Result<R, Error>
where
    T: LinalgRuntimeValue,