Support overriding memory allocation functions
- Let users provide their own memory management functions for XNNPACK
PiperOrigin-RevId: 281355722
diff --git a/bench/softargmax.cc b/bench/softargmax.cc
index cff4023..409525b 100644
--- a/bench/softargmax.cc
+++ b/bench/softargmax.cc
@@ -29,7 +29,7 @@
std::generate(input.begin(), input.end(), std::ref(u8rng));
std::fill(output.begin(), output.end(), 0xA5);
- xnn_status status = xnn_initialize();
+ xnn_status status = xnn_initialize(nullptr /* allocator */);
if (status != xnn_status_success) {
state.SkipWithError("failed to initialize XNNPACK");
return;