Struct KernelMigrator 

pub struct KernelMigrator { /* private fields */ }

Migrator that uses checkpoints for kernel state transfer between GPUs.

This integrates the checkpoint infrastructure with the multi-GPU migration system to enable live migration of persistent kernels.

Example

use ringkernel_core::multi_gpu::{KernelMigrator, MultiGpuBuilder};

let coordinator = MultiGpuBuilder::new().build();
let migrator = KernelMigrator::new(coordinator);

// Migrate kernel from GPU 0 to GPU 1
migrator.migrate_with_checkpoint(&kernel, &mut request).await?;

Implementations

impl KernelMigrator

pub fn new(coordinator: Arc<MultiGpuCoordinator>) -> Self

Create a new kernel migrator with default in-memory storage.

pub fn with_storage( coordinator: Arc<MultiGpuCoordinator>, storage: Arc<dyn CheckpointStorage>, ) -> Self

Create a migrator with custom checkpoint storage.

pub fn migrate_with_checkpoint<K: CheckpointableKernel>( &self, kernel: &K, request: &mut MigrationRequest, ) -> Result<MigrationResult>

Perform a complete migration using checkpoint-based state transfer.

Steps:

1. Quiesce the source kernel (drain pending messages)
2. Create checkpoint of kernel state
3. Transfer checkpoint to target device
4. Restore kernel on target device
5. Update coordinator routing tables

pub fn coordinator(&self) -> &Arc<MultiGpuCoordinator>

Get a reference to the coordinator.

pub fn stats(&self) -> MigrationStatsSnapshot

Get migration statistics snapshot.