spirv-webgpu-transform 0.1.6

Transform SPIRV to be webgpu friendly
Documentation 
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use super::*;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub(super) enum IsNanOrIsInf {
    IsNan,
    IsInf,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub(super) struct NanInfSharedTypeInputs {
    pub uint_id: u32,
    pub ptr_uint_id: u32,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub(super) struct NanInfSharedFunctionInputs {
    pub bool_id: u32,
    pub float_id: u32,
    pub ptr_float_id: u32,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub(super) struct NanInfFunctionType(pub u32);

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub(super) struct NanInfSharedConstants {
    pub uint_23: u32,
    pub uint_255: u32,
    pub uint_8388607: u32,
    pub uint_0: u32,
}

pub(super) fn nan_inf_fn_type_spv(
    ib: &mut u32,
    inputs: NanInfSharedFunctionInputs,
) -> (NanInfFunctionType, Vec<u32>) {
    //
    //    %_function_type = OpTypeFunction %bool %_ptr_Function_float

    let function_type = inc(ib);
    #[rustfmt::skip]
    let spv = vec![
        encode_word(4, SPV_INSTRUCTION_OP_TYPE_FUNCTION), 
            function_type, inputs.bool_id, inputs.ptr_float_id,
    ];

    (NanInfFunctionType(function_type), spv)
}

pub(super) fn nan_inf_shared_constants_spv(
    ib: &mut u32,
    inputs: NanInfSharedTypeInputs,
) -> (NanInfSharedConstants, Vec<u32>) {
    //
    //           %uint_23 = OpConstant %uint 23
    //          %uint_255 = OpConstant %uint 255
    //      %uint_8388607 = OpConstant %uint 8388607
    //            %uint_0 = OpConstant %uint 0

    let uint_23 = inc(ib);
    let uint_255 = inc(ib);
    let uint_8388607 = inc(ib);
    let uint_0 = inc(ib);

    #[rustfmt::skip]
    let spv = vec![
        encode_word(4, SPV_INSTRUCTION_OP_CONSTANT), 
            inputs.uint_id, uint_23, 23,
        encode_word(4, SPV_INSTRUCTION_OP_CONSTANT), 
            inputs.uint_id, uint_255, 255,
        encode_word(4, SPV_INSTRUCTION_OP_CONSTANT), 
            inputs.uint_id, uint_8388607, 8388607,
        encode_word(4, SPV_INSTRUCTION_OP_CONSTANT), 
            inputs.uint_id, uint_0, 0,
    ];

    (
        NanInfSharedConstants {
            uint_23,
            uint_255,
            uint_8388607,
            uint_0,
        },
        spv,
    )
}