From 3695a2b00fb20ab04092a2c4cac9d904ffc58382 Mon Sep 17 00:00:00 2001 From: "zhou.weiguo" Date: Mon, 10 Jun 2024 12:37:27 +0800 Subject: [PATCH] examples: refine tensor dump --- examples/benchmark/benchmark-matmult.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/examples/benchmark/benchmark-matmult.cpp b/examples/benchmark/benchmark-matmult.cpp index 7446568ee..11bd37035 100644 --- a/examples/benchmark/benchmark-matmult.cpp +++ b/examples/benchmark/benchmark-matmult.cpp @@ -39,9 +39,15 @@ typedef struct { } block_q8_0; static inline float ggml_compute_fp16_to_fp32(uint16_t h) { +#if defined(__ARM_NEON) + __fp16 tmp; + memcpy(&tmp, &h, sizeof(uint16_t)); + return (float) tmp; +#else uint16_t tmp; memcpy(&tmp, &h, sizeof(uint16_t)); return (float) tmp; +#endif } static float tensor_sum_elements(const ggml_tensor * tensor) {