diff --git a/include/tim/vx/platform/native.h b/include/tim/vx/platform/native.h index e254db2..ff42e82 100644 --- a/include/tim/vx/platform/native.h +++ b/include/tim/vx/platform/native.h @@ -50,7 +50,7 @@ class NativeExecutable : public IExecutable{ void GetOutput(const std::vector>& th) override; bool Submit(const std::shared_ptr& ref, bool after = true) override; bool Trigger(bool async = false) override; - std::shared_ptr AllocateTensor(const TensorSpec& tensor_spec); + std::shared_ptr AllocateTensor(const TensorSpec& tensor_spec) override; bool Verify() override; protected: diff --git a/src/tim/vx/internal/src/vip/virtual_device_private.h b/src/tim/vx/internal/src/vip/virtual_device_private.h index d3d7d1e..a1fb7b6 100644 --- a/src/tim/vx/internal/src/vip/virtual_device_private.h +++ b/src/tim/vx/internal/src/vip/virtual_device_private.h @@ -21,13 +21,14 @@ * DEALINGS IN THE SOFTWARE. * *****************************************************************************/ -#ifndef _VIP_VIRTUAL_DEVICE_PRIVARE_H +#ifndef _VIP_VIRTUAL_DEVICE_PRIVATE_H #define _VIP_VIRTUAL_DEVICE_PRIVATE_H #include #include #include #include +#include #include #include #include