skip DMD tests on MPS

helmholtz-analytics · Dec 11, 2024 · c7cdf34 · c7cdf34
1 parent 164b7b7
commit c7cdf34
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/heat/decomposition/tests/test_dmd.py b/heat/decomposition/tests/test_dmd.py
@@ -1,12 +1,18 @@
 import os
 import unittest
+import platform
 import numpy as np
 import torch
 import heat as ht
 
 from ...core.tests.test_suites.basic_test import TestCase
 
+# MPS does not support non-float matrix multiplication
+envar = os.getenv("HEAT_TEST_USE_DEVICE", "cpu")
+is_mps = envar == "gpu" and platform.system() == "Darwin"
 
+
+@unittest.skipIf(is_mps, "MPS does not support non-float matrix multiplication")
 class TestDMD(TestCase):
     def test_dmd_setup_and_catch_wrong(self):
         # catch wrong inputs