Add unit tests to ci

Author: Ryo-not-rio

.github/workflows/precommit.yml

@@ -68,6 +68,9 @@ jobs:
       - name: Run smoke tests
         run: docker run --rm toolsolutions-pytorch:latest ./test.sh
 
+      - name: Run unit tests
+        run: docker run --rm toolsolutions-pytorch:latest bash ./run_unit_tests.sh
+
   # This job adds a check named "Precommit-status" that represents overall
   # workflow status and can be used in branch rulesets
   status:

ML-Frameworks/pytorch-aarch64/.dockerignore

@@ -1,4 +1,7 @@
 ComputeLibrary/
 pytorch/
+!pytorch/test
+!pytorch/test/**
+!pytorch/.ci/docker/requirements-ci.txt
 *.log
 .*_build_container_id

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -103,8 +103,11 @@ RUN pip install "$(basename "$TORCH_WHEEL")" --no-deps \
 RUN pip install "$(basename "$TORCH_AO_WHEEL")" --no-deps \
     && rm "$(basename "$TORCH_AO_WHEEL")"
 
-# Setup Examples
+# Setup Examples and tests
 COPY examples/ /home/$DOCKER_USER/
+COPY pytorch/test /home/$DOCKER_USER/pytorch/test
+
+RUN pip install -r requirements-unit.txt
 
 # Llm examples depends on torchchat
 RUN sudo mkdir -p /home/ubuntu/gen_ai_utils/ && \

ML-Frameworks/pytorch-aarch64/examples/requirements-unit.txt

@@ -0,0 +1 @@
+expecttest==0.3.0

ML-Frameworks/pytorch-aarch64/examples/run_unit_tests.sh

@@ -0,0 +1,23 @@
+#!/bin/bash -e
+
+# *******************************************************************************
+# Copyright 2025 Arm Limited and affiliates.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# *******************************************************************************
+
+LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libtcmalloc.so.4 DNNL_DEFAULT_FPMATH_MODE=BF16 /
+    python pytorch/test/test_mkldnn.py
+LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libtcmalloc.so.4 DNNL_DEFAULT_FPMATH_MODE=BF16 /
+    python pytorch/test/test_transformers.py

ML-Frameworks/pytorch-aarch64/examples/test.sh

@@ -31,5 +31,4 @@ done
 
 # Check an example with some of the flags from REAMDE.md > "General optimization
 # guidelines" There is no verbatim example for this
-LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libtcmalloc.so.4  IDEEP_CACHE_MATMUL_REORDERS=1 \
-    LRU_CACHE_CAPACITY=256 DNNL_DEFAULT_FPMATH_MODE=BF16 ${examples[0]}
+LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libtcmalloc.so.4 DNNL_DEFAULT_FPMATH_MODE=BF16 ${examples[0]}