address comments

Author: Ryo-not-rio

.github/workflows/precommit.yml

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

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -107,8 +107,6 @@ RUN pip install "$(basename "$TORCH_AO_WHEEL")" --no-deps \
 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/ && \
     cd /home/ubuntu/gen_ai_utils/ && \

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

@@ -17,7 +17,5 @@
 # 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
+OMP_NUM_THREADS=16 python pytorch/test/test_mkldnn.py
+OMP_NUM_THREADS=16 python pytorch/test/test_transformers.py

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

@@ -4,6 +4,7 @@
 certifi==2024.8.30
 charset-normalizer~=3.4.0
 datasets~=3.4.1
+expecttest==0.3.0 # From unit tests
 filelock~=3.16.1
 fsspec==2024.9.0
 huggingface_hub==0.27.0