diff --git a/backends/arm/runtime/EthosUBackend.cpp b/backends/arm/runtime/EthosUBackend.cpp index c51ea9cc18c..940f07f8a18 100644 --- a/backends/arm/runtime/EthosUBackend.cpp +++ b/backends/arm/runtime/EthosUBackend.cpp @@ -10,6 +10,7 @@ * ethos-u-core-driver for hardware interaction. */ +#include #include #include @@ -282,7 +283,10 @@ class EthosUBackend final : public ::executorch::runtime::BackendInterface { // constant weight data, then scratch (which contains input and output) // scratch is written above in this function. uint64_t bases[2] = { - (uint64_t)handles.weight_data, (uint64_t)handles.scratch_data}; + static_cast( + reinterpret_cast((handles.weight_data))), + static_cast( + reinterpret_cast((handles.scratch_data)))}; size_t bases_size[2] = { handles.weight_data_size, handles.scratch_data_size}; int result = 0;