Docker env for Arm® Ethos™-U55 Port

* Adds ethosu as an extra to /python/gen_requirements.py

Change-Id: I2162b40f82241fd013643cbfa8847b60d7f4f5a1

docker/Dockerfile.ci_cpu

@@ -118,4 +118,4 @@ COPY install/ubuntu_install_vela.sh /install/ubuntu_install_vela.sh
 RUN bash /install/ubuntu_install_vela.sh
 
 # Update PATH
-ENV PATH /opt/arm/bin:/opt/arm/gcc-arm-none-eabi/bin:$PATH
+ENV PATH /opt/arm/gcc-arm-none-eabi/bin:$PATH

docker/install/ubuntu_install_ethosu_driver_stack.sh

@@ -24,6 +24,7 @@ fvp_dir="/opt/arm/FVP_Corstone_SSE-300_Ethos-U55"
 cmake_dir="/opt/arm/cmake"
 ethosu_dir="/opt/arm/ethosu"
 ethosu_driver_ver="21.05"
+cmsis_ver="5.7.0"
 
 mkdir -p /opt/arm
 
@@ -75,7 +76,6 @@ curl --retry 64 -sSL ${gcc_arm_url} | tar -C /opt/arm/gcc-arm-none-eabi --strip-
 export PATH="/opt/arm/gcc-arm-none-eabi/bin:${PATH}"
 
 # Clone Arm(R) Ethos(TM)-U NPU driver stack
-cd /
 mkdir "${ethosu_dir}"
 cd "${ethosu_dir}"
 git clone "https://review.mlplatform.org/ml/ethos-u/ethos-u-core-driver" core_driver
@@ -91,4 +91,4 @@ git checkout tags/${ethosu_driver_ver}
 cd "${ethosu_dir}"
 git clone "https://github.com/ARM-software/CMSIS_5.git" cmsis
 cd cmsis
-git checkout -f tags/5.7.0
+git checkout -f tags/${cmsis_ver}

python/gen_requirements.py

@@ -75,6 +75,16 @@
             ],
         ),
     ),
+    # Provide support for Arm(R) Ethos(TM)-U NPU.
+    (
+        "ethosu",
+        (
+            "Requirements for using Arm(R) Ethos(TM)-U NPU",
+            [
+                "ethos-u-vela",
+            ],
+        ),
+    ),
     # Relay frontends.
     (
         "importer-caffe2",
@@ -205,6 +215,7 @@
         "docutils",
         "<0.17",
     ),  # Work around https://github.com/readthedocs/sphinx_rtd_theme/issues/1115
+    ("ethos-u-vela", "==2.1.1"),
     ("future", None),
     ("image", None),
     ("matplotlib", None),