python3Packages.tinygrad: fix GPU tests (#524737)

2026-05-27 15:22:12 +00:00
parent 8850d52db0 d295eb6cb4
commit 5deb415c63
1 changed files with 9 additions and 0 deletions
@@ -308,6 +308,15 @@ buildPythonPackage (finalAttrs: {
    gpuCheck = tinygrad.overridePythonAttrs (old: {
      requiredSystemFeatures = [ "cuda" ];

+      disabledTests = (old.disabledTests or [ ]) ++ [
+        # Require internet access
+        "TestWhisper"
+        "test_hevc_decode"
+
+        # AssertionError: Not equal to tolerance
+        "test_svd_general"
+      ];
+
      pytestFlags = (old.pytestFlags or [ ]) ++ [
        # When running in parallel, with GPU support, some tests become flaky:
        # RuntimeError: Wait timeout: 30000 ms! (the signal is not set to 153, but 151)