examples: refine tensor dump

zhouwg · Jun 10, 2024 · 3695a2b · 3695a2b
1 parent 2c96a5c
commit 3695a2b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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) {