From c1dd25137aa1ce80e7e2cb27472d0fa16352fbbc Mon Sep 17 00:00:00 2001 From: Aleksandr Voron Date: Fri, 20 Dec 2024 18:15:03 +0100 Subject: [PATCH] [ARM][FORK] Resolve float32_t type --- src/common/memory_tracking.hpp | 4 ++++ src/cpu/acl/matmul/acl_lowp_matmul.cpp | 6 ++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/src/common/memory_tracking.hpp b/src/common/memory_tracking.hpp index c5b1cba832c..2b37f9fc05a 100644 --- a/src/common/memory_tracking.hpp +++ b/src/common/memory_tracking.hpp @@ -301,6 +301,10 @@ enum { key_wino_U, key_wino_V, key_wino_M, + key_decompression_scales, + key_decompression_zero_points, + key_src_quantized, + key_src_dequantized_scales, // These two keys should always be the last ones, // even though they are not in alphabetical order key_nested, diff --git a/src/cpu/acl/matmul/acl_lowp_matmul.cpp b/src/cpu/acl/matmul/acl_lowp_matmul.cpp index c0c278ffd08..647a4c65322 100644 --- a/src/cpu/acl/matmul/acl_lowp_matmul.cpp +++ b/src/cpu/acl/matmul/acl_lowp_matmul.cpp @@ -16,6 +16,8 @@ #include "cpu/acl/matmul/acl_lowp_matmul.hpp" +#include "src/cpu/CpuTypes.h" + namespace dnnl { namespace impl { namespace cpu { @@ -203,11 +205,11 @@ status_t acl_lowp_matmul_t::pd_t::init_scratchpad( const memory_desc_wrapper dst_d(&dst_md_); if (almc_.use_dst_acc) { scratchpad.book(memory_tracking::names::key_matmul_dst_in_acc_dt, - dst_d.nelems(), sizeof(float32_t)); + dst_d.nelems(), sizeof(arm_compute::float32_t)); } if (almc_.use_cast_acc) { scratchpad.book(memory_tracking::names::key_matmul_dst_cast_acc, - dst_d.nelems(), sizeof(float32_t)); + dst_d.nelems(), sizeof(arm_compute::float32_t)); } return status::success; }