Struct GpuState 

pub struct GpuState {
    pub device: Arc<Device>,
    pub queue: Arc<Queue>,
    pub nworld: usize,
    pub nq: usize,
    pub nv: usize,
    pub q_buffer: Buffer,
    pub v_buffer: Buffer,
    pub ctrl_buffer: Buffer,
    pub qdd_buffer: Buffer,
    pub q_staging: Buffer,
    pub v_staging: Buffer,
}

GPU-backed batch simulation state.

Stores state for nworld parallel environments on GPU memory.

Fields

device: Arc<Device>

queue: Arc<Queue>

nworld: usize

nq: usize

nv: usize

q_buffer: Buffer

v_buffer: Buffer

ctrl_buffer: Buffer

qdd_buffer: Buffer

q_staging: Buffer

v_staging: Buffer

Implementations

impl GpuState

pub fn new( device: Arc<Device>, queue: Arc<Queue>, model: &Model, nworld: usize, ) -> Self

Create GPU buffers for batch simulation.

pub fn upload_states(&self, states: &[State])

Upload states from CPU to GPU.

pub async fn download_states(&self) -> Result<(Vec<f32>, Vec<f32>), String>

Download states from GPU to CPU.