summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--pkgs/development/python-modules/torch/gpu-checks.nix2
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/torch/gpu-checks.nix b/pkgs/development/python-modules/torch/gpu-checks.nix
index d01fffe45cb0..55a4b45f7152 100644
--- a/pkgs/development/python-modules/torch/gpu-checks.nix
+++ b/pkgs/development/python-modules/torch/gpu-checks.nix
@@ -13,7 +13,7 @@ let
torch,
cudaPackages,
}:
- cudaPackages.writeGpuPythonTest
+ cudaPackages.writeGpuTestPython
{
inherit feature;
libraries = [ torch ];