wonnx 0.5.1

Wonnx is an ONNX runtime based on wgpu aimed at being a universal GPU runtime, written in Rust.
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

{# 
// Operations usually work with a single scalar data type (typically f32). This data type is set by the compiler as the
// 'scalar_type' variable. Here we define several other useful data types that shader code can use to make it more portable.
#}
alias Scalar = {{ scalar_type }};
alias Vec3 = vec3<{{ scalar_type }}>;
alias Vec4 = vec4<{{ scalar_type }}>;

struct Array {
	data: array<Scalar>
};

struct ArrayVector {
	data: array<Vec4>
};

{# 
// WGSL only supports matrixes for floating point types at this point 
#}
{% if scalar_type_is_float %}
	alias Mat3x3 = mat3x3<{{ scalar_type }}>;
	alias Mat4x4 = mat4x4<{{ scalar_type }}>;
	alias Mat4x3 = mat4x3<{{ scalar_type }}>;

	struct ArrayMatrix {
		data: array<Mat4x4>
	};

	struct ArrayMatrix3 {
		data: array<Mat3x3>
	};
{% endif %}