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
28
29
30
31
32
33
34
35
36
37
38

use std::ffi::{CStr, CString};

mod preprocess;
mod regex;
mod replace;

type ByteSliceOptionalArguments = Option<Box<[Option<Box<[u8]>>]>>;

pub struct PtxJITCompiler {
    ptx_slices: Box<[PtxElement]>,
    last_arguments: ByteSliceOptionalArguments,
    last_ptx: CString,
}

pub enum PtxJITResult<'s> {
    Cached(&'s CStr),
    Recomputed(&'s CStr),
}

enum PtxLoadWidth {
    B1,
    B2,
    B4,
    B8,
}

enum PtxElement {
    CopiedSource {
        ptx: Box<[u8]>,
    },
    ConstLoad {
        ptx: Box<[u8]>,
        parameter_index: usize,
        byte_offset: usize,
        load_width: PtxLoadWidth,
        registers: Box<[Box<[u8]>]>,
    },
}