fixup! fixup! fixup! fixup! python3Packages.triton.tests.axpy-cuda: init

NixOS · Oct 15, 2024 · 842b3c0 · 842b3c0
1 parent 102bad2
commit 842b3c0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkgs/development/python-modules/triton/default.nix b/pkgs/development/python-modules/triton/default.nix
@@ -181,6 +181,8 @@ buildPythonPackage {
     pname = "triton-pytest";
     inherit (triton) version src;
 
+    requiredSystemFeatures = [ "cuda" ];
+
     nativeBuildInputs = [
       (python.withPackages (ps: [
         ps.scipy
@@ -189,15 +191,16 @@ buildPythonPackage {
       ]))
     ];
 
-    nativeCheckInputs = [ pytestCheckHook ];
     dontBuild = true;
+    nativeCheckInputs = [ pytestCheckHook ];
+
     doCheck = true;
-    requiredSystemFeatures = [ "cuda" ];
 
     preCheck = ''
       cd python/test/unit
       export HOME=$TMPDIR
     '';
+    checkPhase = "pytestCheckPhase";
 
     installPhase = "touch $out";
   };