Skip to main content

cuda_runtime_engine_registration

Function cuda_runtime_engine_registration 

Source
pub fn cuda_runtime_engine_registration(
    engine: &CudaEngine,
) -> Result<EngineRegistration, RuntimeConfigError>
Expand description

Build the runtime registration for engine.

§Examples

use tenferro_gpu::cuda::runtime_adapter::cuda_runtime_engine_registration;
use tenferro_gpu::cuda::{CudaDeviceId, CudaEngine};
use tenferro_runtime::Runtime;

let engine = CudaEngine::open(CudaDeviceId::from_ordinal(0))?;
let mut builder = Runtime::builder();
builder.register_engine(cuda_runtime_engine_registration(&engine)?)?;
let runtime = builder.build()?;
let _snapshot = runtime.snapshot()?;

§Errors

RuntimeConfigError when the provider identity, hardware or storage class, or the registration itself cannot be built.