python312Packages.torch: fix tests
This commit is contained in:
@@ -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 = [
|
||||
|
||||
Reference in New Issue
Block a user