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