cpu: aarch64: deconv: fix ACL resource allocation

oneapi-src · Nov 18, 2022 · f02e6f3 · f02e6f3
1 parent 2ba2523
commit f02e6f3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/cpu/ref_deconvolution.hpp b/src/cpu/ref_deconvolution.hpp
@@ -1,5 +1,6 @@
 /*******************************************************************************
 * Copyright 2018-2022 Intel Corporation
+* Copyright 2022 Arm Ltd. and affiliates
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -379,6 +380,14 @@ struct ref_deconvolution_bwd_data_t : public primitive_t {
         return pd()->conv_pd_->create_primitive(conv_p_, engine);
     }
 
+#if DNNL_AARCH64 && DNNL_AARCH64_USE_ACL
+    status_t create_resource(
+            engine_t *engine, resource_mapper_t &mapper) const override {
+        CHECK(conv_p_->create_resource(engine, mapper));
+        return status::success;
+    }
+#endif
+
     status_t execute(const exec_ctx_t &ctx) const override;
 
 private: