Method get_or_create

cake-core/src/backends/metal/mod.rs:62–92

(
        &self,
        device: &candle_core::MetalDevice,
        kernel_name: &'static str,
    )

    }

    fn get_or_create(
        &self,
        device: &candle_core::MetalDevice,
        kernel_name: &'static str,
    ) -> Result<candle_metal_kernels::metal::ComputePipeline> {
        if let Ok(cache) = self.pipelines.read() {
            if let Some(pipeline) = cache.get(kernel_name) {
                return Ok(pipeline.clone());
            }
        }
        let _guard = self.compile_lock.lock().map_err(|e| candle_core::Error::Msg(format!("compile lock: {e}")))?;
        if let Ok(cache) = self.pipelines.read() {
            if let Some(pipeline) = cache.get(kernel_name) {
                return Ok(pipeline.clone());
            }
        }
        let lib = device.new_library_with_source(FUSED_OPS_MSL, None)
            .map_err(|e| candle_core::Error::Msg(format!("metal shader compile: {e}")))?;
        let mut cache = self.pipelines.write().map_err(|e| candle_core::Error::Msg(format!("pipeline write lock: {e}")))?;
        for &name in ALL_KERNELS {
            if cache.contains_key(name) { continue; }
            if let Ok(func) = lib.get_function(name, None) {
                if let Ok(pipeline) = device.new_compute_pipeline_state_with_function(&func) {
                    cache.insert(name, pipeline);
                }
            }
        }
        cache.get(kernel_name).cloned().ok_or_else(|| {
            candle_core::Error::Msg(format!("metal kernel not found: {kernel_name}"))
        })
    }
}

static PIPELINE_CACHE: std::sync::LazyLock<PipelineCache> = std::sync::LazyLock::new(PipelineCache::new);