with_einsum_runtime

tenferro_internal_ad_surface::runtime::dispatch

Function with_einsum_runtime 

pub fn with_einsum_runtime<T, R>(
    op: &'static str,
    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: EinsumRuntimeValue,