Merge pull request #407 from puneetmatharu/patch-failing-pytorch-tests

Remove PyTorch float16 testing with bf16 fpmath mode

Author: puneetmatharu

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

@@ -18,13 +18,15 @@
 # *******************************************************************************
 
 if [[ "${ONEDNN_DEFAULT_FPMATH_MODE:-}" == "BF16" ]]; then
-    OMP_NUM_THREADS=16 python -m unittest pytorch/test/test_mkldnn.py -k lower_precision -k bf16 -k bfloat16 -k float16
+    # Run bfloat16 tests but carefully ignore float16
+    OMP_NUM_THREADS=16 python3 -m pytest -q pytorch/test/test_mkldnn.py -k 'bfloat16 or ((lower_precision or bf16) and not float16)'
 else
-    OMP_NUM_THREADS=16 python -m unittest pytorch/test/test_mkldnn.py
+    OMP_NUM_THREADS=16 python3 -m pytest -q pytorch/test/test_mkldnn.py
 fi
 
 if [[ "${ONEDNN_DEFAULT_FPMATH_MODE:-}" == "BF16" ]]; then
-    OMP_NUM_THREADS=16 python -m unittest pytorch/test/test_transformers.py -k bfloat16 -k float16
+    # Run bfloat16 tests but carefully ignore float16
+    OMP_NUM_THREADS=16 python3 -m pytest -q pytorch/test/test_transformers.py -k 'bfloat16 or ((lower_precision or bf16) and not float16)'
 else
-    OMP_NUM_THREADS=16 python -m unittest pytorch/test/test_transformers.py
+    OMP_NUM_THREADS=16 python3 -m pytest -q pytorch/test/test_transformers.py
 fi

ML-Frameworks/pytorch-aarch64/requirements.txt

@@ -22,6 +22,7 @@ pyaml~=24.9.0
 python-dateutil~=2.9.0.post0
 pytz==2024.2
 PyYAML~=6.0.2
+pytest~=8.4.2
 regex==2024.9.11
 requests~=2.32.3
 safetensors~=0.4.5