From 61ad8d91ccaf4c1e6c3e4456d7a137eaeb107a06 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 5 Jan 2024 14:31:39 +0100 Subject: [PATCH] Fix the rebase. --- candle-examples/examples/quantized/main.rs | 4 ++-- candle-metal-kernels/src/unary.metal | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/candle-examples/examples/quantized/main.rs b/candle-examples/examples/quantized/main.rs index 4525ba42..34c44233 100644 --- a/candle-examples/examples/quantized/main.rs +++ b/candle-examples/examples/quantized/main.rs @@ -370,7 +370,7 @@ fn main() -> anyhow::Result<()> { for (_, tensor) in model.tensor_infos.iter() { let elem_count = tensor.shape.elem_count(); total_size_in_bytes += - elem_count * tensor.ggml_dtype.type_size() / tensor.ggml_dtype.blck_size(); + elem_count * tensor.ggml_dtype.type_size() / tensor.ggml_dtype.block_size(); } println!( "loaded {:?} tensors ({}) in {:.2}s", @@ -387,7 +387,7 @@ fn main() -> anyhow::Result<()> { for (_, tensor) in model.tensors.iter() { let elem_count = tensor.shape().elem_count(); total_size_in_bytes += - elem_count * tensor.dtype().type_size() / tensor.dtype().blck_size(); + elem_count * tensor.dtype().type_size() / tensor.dtype().block_size(); } println!( "loaded {:?} tensors ({}) in {:.2}s", diff --git a/candle-metal-kernels/src/unary.metal b/candle-metal-kernels/src/unary.metal index bffe8e60..7add58fd 100644 --- a/candle-metal-kernels/src/unary.metal +++ b/candle-metal-kernels/src/unary.metal @@ -135,6 +135,7 @@ BFLOAT_UNARY_OP(neg) BFLOAT_UNARY_OP(exp) BFLOAT_UNARY_OP(log) BFLOAT_UNARY_OP(gelu) +BFLOAT_UNARY_OP(abs) BFLOAT_UNARY_OP(ceil) BFLOAT_UNARY_OP(floor) BFLOAT_UNARY_OP(round) @@ -143,7 +144,6 @@ BFLOAT_UNARY_OP(erf) BFLOAT_UNARY_OP(tanh) BFLOAT_UNARY_OP(recip) BFLOAT_UNARY_OP(relu) -BFLOAT_UNARY_OP(abs) UNARY(id, bfloat, copy_bf16, copy_bf16_strided) #endif