3 |
4 | 5 | A web-first, cross-platform ML developer toolkit 6 |
7 |8 |
,
75 | }
76 |
77 | impl {
78 | fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
79 | write!(f, "array<{}>", P::render_type())
80 | }
81 | }
82 |
--------------------------------------------------------------------------------
/crates/ratchet-core/src/gpu/wgsl/dtype.rs:
--------------------------------------------------------------------------------
1 | use half::f16;
2 | use std::fmt::{Debug, Display};
3 |
4 | /// Supported data types in WGSL.
5 | ///
6 | /// This can be mapped to and from the Ratchet DType.
7 | pub trait WgslDType: Debug + Display + Default + Copy + num_traits::Num + num_traits::Zero {
8 | const DT: &'static str;
9 | const MIN: Self;
10 |
11 | fn render(&self) -> String;
12 | }
13 | //RENDER IS CONFUSING HERE
14 |
15 | impl WgslDType for f32 {
16 | const DT: &'static str = "f32";
17 | const MIN: Self = -3e10; //ranges for wgsl and rust are diff
18 |
19 | fn render(&self) -> String {
20 | format!("{}f", self)
21 | }
22 | }
23 |
24 | impl WgslDType for f16 {
25 | const DT: &'static str = "f16";
26 | const MIN: Self = f16::MIN;
27 |
28 | fn render(&self) -> String {
29 | format!("{}h", self)
30 | }
31 | }
32 |
33 | impl WgslDType for i32 {
34 | const DT: &'static str = "i32";
35 | const MIN: Self = i32::MIN;
36 |
37 | fn render(&self) -> String {
38 | format!("{}i", self)
39 | }
40 | }
41 |
42 | impl WgslDType for u32 {
43 | const DT: &'static str = "u32";
44 | const MIN: Self = u32::MIN;
45 |
46 | fn render(&self) -> String {
47 | format!("{}u", self)
48 | }
49 | }
50 |
--------------------------------------------------------------------------------
/crates/ratchet-core/src/gpu/wgsl/kernel_binding.rs:
--------------------------------------------------------------------------------
1 | use crate::{Ident, RenderFragment};
2 | use inline_wgsl::wgsl;
3 |
4 | #[derive(Debug, PartialEq, Eq)]
5 | pub(crate) enum BindingType {
6 | Storage,
7 | Uniform,
8 | }
9 |
10 | #[derive(Debug, Copy, Clone)]
11 | pub(crate) enum BindingMode {
12 | ReadOnly,
13 | ReadWrite,
14 | }
15 |
16 | impl BindingMode {
17 | pub fn as_str(&self) -> &'static str {
18 | match self {
19 | BindingMode::ReadOnly => "read",
20 | BindingMode::ReadWrite => "read_write",
21 | }
22 | }
23 | }
24 |
25 | #[derive(Debug, derive_new::new)]
26 | pub struct KernelBinding {
27 | name: Ident,
28 | group: usize,
29 | binding: usize,
30 | ty: BindingType,
31 | mode: BindingMode,
32 | accessor: String,
33 | }
34 |
35 | impl From