python3Packages.jaxlibWithCuda: init at 0.1.75 (#157055)

* python3Packages.jaxlibWithCuda: init at 0.1.75 * python3Packages.jaxlibWithoutCuda: init at 0.1.75
2022-01-28 11:04:10 -08:00
parent 5b958f0c22
commit 65a33fec2a
1 changed files with 8 additions and 0 deletions
@@ -4127,6 +4127,14 @@ in {

  jaxlib = self.jaxlib-build;

+  jaxlibWithCuda = self.jaxlib-build.override {
+    cudaSupport = true;
+  };
+
+  jaxlibWithoutCuda = self.jaxlib-build.override {
+    cudaSupport = false;
+  };
+
  JayDeBeApi = callPackage ../development/python-modules/JayDeBeApi { };

  jc = callPackage ../development/python-modules/jc { };