-
-
Notifications
You must be signed in to change notification settings - Fork 89
Description
Comment:
We've had to patch out many things in the HERMETIC CUDA path.
Howover, in discussions with upstream tensorflow/tensorflow#86413 @ybaturina suggests
I think the case you've described applies to local CUDA installation only. In this situation I can suggest commenting out cc_import for libnvrtc-builtins.so when LOCAL_CUDA_PATH is set. It can be done somewhere at these lines - https://github.com/openxla/xla/blob/main/third_party/gpus/cuda/hermetic/cuda_redist_init_repositories.bzl#L188-L195. You can add special type of comments around that cc_import, and replace them by real comment symbols when LOCAL_CUDA_PATH is set (for cuda_nvrtc repo only). This will preserve hermetic CUDA state when non-local installation is used.
tensorflow/tensorflow#86413 (comment)
It may be worthwhile for us to revisit this!