python312Packages.torch: fix tests

This commit is contained in:
Gaetan Lepage
2025-05-05 22:19:53 +02:00
parent 39836de377
commit 5bc385e73e
2 changed files with 2 additions and 4 deletions
@@ -3,10 +3,9 @@
feature,
libraries,
versionAttr,
pythonPackages,
}:
(cudaPackages.writeGpuTestPython.override { python3Packages = pythonPackages; })
cudaPackages.writeGpuTestPython
{
inherit feature;
inherit libraries;
@@ -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 = [