diff options
| author | Gonzalo <456459+grzuy@users.noreply.github.com> | 2023-12-29 19:56:21 -0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-29 23:56:21 +0100 |
| commit | 87d7f81b438db6a1696f2ac79606b40e61d448e8 (patch) | |
| tree | 31f3fb7a25091b0a9076d1aa19d510c5b3d9b044 /candle-metal-kernels/src/lib.rs | |
| parent | 4373534d59d3a6357aef0b3f35a247f695f4700a (diff) | |
| download | candle-87d7f81b438db6a1696f2ac79606b40e61d448e8.tar.gz candle-87d7f81b438db6a1696f2ac79606b40e61d448e8.tar.bz2 candle-87d7f81b438db6a1696f2ac79606b40e61d448e8.zip | |
Metal: more u8/u32 (#1502)
* Adds more metal u8
* Metal: more u32
Diffstat (limited to 'candle-metal-kernels/src/lib.rs')
| -rw-r--r-- | candle-metal-kernels/src/lib.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-metal-kernels/src/lib.rs b/candle-metal-kernels/src/lib.rs index 7b0084d9..d080ef52 100644 --- a/candle-metal-kernels/src/lib.rs +++ b/candle-metal-kernels/src/lib.rs @@ -131,6 +131,8 @@ macro_rules! ops{ pub const HALF: Kernel = Kernel(concat!(stringify!($name), "_f16")); pub const BFLOAT: Kernel = Kernel(concat!(stringify!($name), "_bf16")); pub const I64: Kernel = Kernel(concat!(stringify!($name), "_i64")); + pub const U32: Kernel = Kernel(concat!(stringify!($name), "_u32")); + pub const U8: Kernel = Kernel(concat!(stringify!($name), "_u8")); } )+ pub mod copy { @@ -153,6 +155,8 @@ macro_rules! ops{ pub const HALF: Kernel = Kernel(concat!(stringify!($name), "_f16_strided")); pub const BFLOAT: Kernel = Kernel(concat!(stringify!($name), "_bf16_strided")); pub const I64: Kernel = Kernel(concat!(stringify!($name), "_i64_strided")); + pub const U32: Kernel = Kernel(concat!(stringify!($name), "_u32_strided")); + pub const U8: Kernel = Kernel(concat!(stringify!($name), "_u8_strided")); } )+ pub mod copy { |