mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-24 22:53:42 +00:00
be52722509
* python3Packages.jaxlib: rename to `jaxlib-bin` Refactoring `jaxlib` to have a similar structure to `tensorflow` with the 'bin' and 'build' options. * python3Packages.jaxlib: init the 'build' variant at 0.1.75 Similar to `tensorflow-build`, now there's an option to build `jaxlib` using Nix-provided environment and dependencies. * python3Packages.jax: 0.2.24 -> 0.2.26 * Addressed review comments. * Fixed `cudaSupport` missing property on some arches. * Unified the versions of CUDA-related packages with TF. Co-authored-by: Samuel Ainsworth <skainsworth@gmail.com>
13 lines
562 B
Diff
13 lines
562 B
Diff
diff --git a/jax/experimental/compilation_cache/file_system_cache.py b/jax/experimental/compilation_cache/file_system_cache.py
|
|
index b85969de..92acd523 100644
|
|
--- a/jax/experimental/compilation_cache/file_system_cache.py
|
|
+++ b/jax/experimental/compilation_cache/file_system_cache.py
|
|
@@ -33,6 +33,7 @@ class FileSystemCache(CacheInterface):
|
|
path_to_key = os.path.join(self._path, key)
|
|
if os.path.exists(path_to_key):
|
|
with open(path_to_key, "rb") as file:
|
|
+ os.utime(file.fileno())
|
|
return file.read()
|
|
else:
|
|
return None
|