From 67e11f7bce40d72e0dda97cf658a3c3ee600c1eb Mon Sep 17 00:00:00 2001 From: Mauricio Briceno Date: Wed, 5 May 2021 12:47:28 +0200 Subject: weight_compressor: added mlw_reorder_encode - Moves reordering to C - Runtime is greatly minimized for encoding weights Change-Id: Ifff01e7b1ea6d5cec68310a155c3b80aa1a38545 Signed-off-by: Mauricio Briceno --- setup.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'setup.py') diff --git a/setup.py b/setup.py index 8cf61951..d2137437 100644 --- a/setup.py +++ b/setup.py @@ -18,6 +18,7 @@ import os import re +import numpy as np from setuptools import Extension from setuptools import find_namespace_packages from setuptools import setup @@ -42,6 +43,7 @@ with open(os.path.join(this_directory, "README.md"), encoding="utf-8") as f: mlw_module = Extension( "ethosu.mlw_codec", ["ethosu/mlw_codec/mlw_encode.c", "ethosu/mlw_codec/mlw_decode.c", "ethosu/mlw_codec/mlw_codecmodule.c"], + include_dirs=[np.get_include()], ) setup( -- cgit v1.2.1