icicle-cuda-runtime 1.3.0

Ingonyama's Rust wrapper of CUDA runtime
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::memory::CudaMemPool;
use crate::stream::CudaStream;

/// Properties of the device used in icicle functions.
#[repr(C)]
#[derive(Debug, Clone)]
pub struct DeviceContext<'a> {
    /// Stream to use. Default value: 0.
    pub stream: &'a CudaStream, // Assuming the type is provided by a CUDA binding crate

    /// Index of the currently used GPU. Default value: 0.
    pub device_id: usize,

    /// Mempool to use. Default value: 0.
    pub mempool: CudaMemPool, // Assuming the type is provided by a CUDA binding crate
}

pub fn get_default_device_context() -> DeviceContext<'static> {
    static default_stream: CudaStream = CudaStream {
        handle: std::ptr::null_mut(),
    };
    DeviceContext {
        stream: &default_stream,
        device_id: 0,
        mempool: 0,
    }
}