Cherry-pick the ptxas binary

dchaley · Jun 2, 2024 · a9ff104 · a9ff104
1 parent ad79db4
commit a9ff104
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/tensorflow/tools/dockerfiles/dockerfiles/gpu.Dockerfile b/tensorflow/tools/dockerfiles/dockerfiles/gpu.Dockerfile
@@ -71,6 +71,8 @@ ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA}/targets/x86_64-linux/lib:/usr/local/
 RUN echo "/usr/local/cuda/compat" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
     && ldconfig
 
+COPY partials/ubuntu/ptxas.binary.x86_64 /usr/bin/ptxas
+
 # See http://bugs.python.org/issue19846
 ENV LANG C.UTF-8
 

diff --git a/tensorflow/tools/dockerfiles/partials/ubuntu/nvidia.partial.Dockerfile b/tensorflow/tools/dockerfiles/partials/ubuntu/nvidia.partial.Dockerfile
@@ -47,3 +47,5 @@ ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA}/targets/x86_64-linux/lib:/usr/local/
 # dynamic linker run-time bindings
 RUN echo "/usr/local/cuda/compat" > /etc/ld.so.conf.d/z-cuda-stubs.conf \
     && ldconfig
+
+COPY partials/ubuntu/ptxas.binary.x86_64 /usr/bin/ptxas
diff --git a/tensorflow/tools/dockerfiles/partials/ubuntu/ptxas.binary.x86_64 b/tensorflow/tools/dockerfiles/partials/ubuntu/ptxas.binary.x86_64