pub fn with_einsum_runtime<T: EinsumRuntimeValue, R>(
op: &'static str,
cpu: impl FnOnce(&mut CpuContext) -> Result<R>,
cuda: impl FnOnce(&mut CudaContext) -> Result<R>,
rocm: impl FnOnce(&mut RocmContext) -> Result<R>,
) -> Result<R>