Unverified Commit 246654b4 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python312Packages.torch: fix tests (#404474)

parents 5a837cb8 5bc385e7

pkgs/development/python-modules/torch/tests/mk-runtime-check.nix

+1 −2
Original line number Diff line number Diff line
@@ -3,10 +3,9 @@ 
  feature,

  libraries,

  versionAttr,

  pythonPackages,

}:



(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })

cudaPackages.writeGpuTestPython

  {

    inherit feature;

    inherit libraries;

pkgs/development/python-modules/torch/tests/mk-torch-compile-check.nix

+1 −2
Original line number Diff line number Diff line
@@ -4,13 +4,12 @@ 
  lib,

  libraries,

  name ? if feature == null then "torch-compile-cpu" else "torch-compile-${feature}",

  pythonPackages,

  stdenv,

}:

let

  deviceStr = if feature == null then "" else '', device="cuda"'';

in

(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })

cudaPackages.writeGpuTestPython

  {

    inherit name feature libraries;

    makeWrapperArgs = [