add python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07:00
parent 580776559b
commit 4fda5fbdf9
36 changed files with 220 additions and 35 deletions
@@ -400,4 +400,4 @@ class TestLoad(mlx_tests.MLXTestCase):


 if __name__ == "__main__":
-    unittest.main()
+    mlx_tests.MLXTestRunner()