MCPcopy Create free account
hub / github.com/evilsocket/cake / device

Method device

cake-core/src/backends/rocm/mod.rs:357–357  ·  view source on GitHub ↗
(&self)

Source from the content-addressed store, hash-verified

355impl ComputeBackend for RocmBackend {
356 fn name(&self) -> &str { "rocm" }
357 fn device(&self) -> &Device { &self.device }
358
359 fn preprocess_linear_weight(&self, weight: &Tensor) -> Result<Tensor> {
360 // Pre-ensure contiguous layout at load time to avoid runtime copies.

Callers 2

attentionMethod · 0.45
stable_softplusMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected