Skip to content

Commit

Permalink
Remove PTX for all but newest GPU architecture.
Browse files Browse the repository at this point in the history 
CUBINs are compatible within the same major architecture, which means we only need to ship PTX to JIT for future architectures.

PiperOrigin-RevId: 315419950
Change-Id: I3eddcadcbbfabb3694e56432a5f257e8f66c5835
  • Loading branch information
@chsigg @tensorflower-gardener
chsigg authored and tensorflower-gardener committed Jun 9, 2020
1 parent 0266394 commit cf1b6b3
Show file tree
Hide file tree
Showing 25 changed files with 25 additions and 25 deletions.
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/common_win.bat
View file
Expand Up @@ -62,7 +62,7 @@ IF NOT DEFINED TF_CUDA_VERSION (
SET TF_CUDA_VERSION=10.1
)
SET TF_CUDNN_VERSION=7
SET TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
SET TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
SET CUDA_TOOLKIT_PATH=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v%TF_CUDA_VERSION%
SET CUDNN_INSTALL_PATH=C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v%TF_CUDA_VERSION%
SET PATH=%CUDA_TOOLKIT_PATH%\extras\CUPTI\libx64;%PATH%
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_pip_on_cpu/build.sh
View file
Expand Up @@ -34,7 +34,7 @@ export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python3.6)
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/nightly_release.sh
View file
Expand Up @@ -31,7 +31,7 @@ export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/nonpip.sh
View file
Expand Up @@ -36,7 +36,7 @@ export PYTHON_BIN_PATH=$(which python2.7)
export TF2_BEHAVIOR=1
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/nonpip_v1.sh
View file
Expand Up @@ -35,7 +35,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python2.7)
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/pip.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py2_full/pip_v1.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/nightly_release.sh
View file
Expand Up @@ -31,7 +31,7 @@ export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/nonpip.sh
View file
Expand Up @@ -36,7 +36,7 @@ export PYTHON_BIN_PATH=$(which python3.5)
export TF2_BEHAVIOR=1
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/nonpip_v1.sh
View file
Expand Up @@ -35,7 +35,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python3.5)
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/pip.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/pip_v1.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/nightly_release.sh
View file
Expand Up @@ -31,7 +31,7 @@ export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/nonpip.sh
View file
Expand Up @@ -36,7 +36,7 @@ export PYTHON_BIN_PATH=$(which python3.6)
export TF2_BEHAVIOR=1
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/nonpip_v1.sh
View file
Expand Up @@ -35,7 +35,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python3.6)
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/pip.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py36_full/pip_v1.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nightly_release.sh
View file
Expand Up @@ -31,7 +31,7 @@ export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nonpip.sh
View file
Expand Up @@ -36,7 +36,7 @@ export PYTHON_BIN_PATH=$(which python3.7)
export TF2_BEHAVIOR=1
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/nonpip_v1.sh
View file
Expand Up @@ -35,7 +35,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which python3.7)
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/pip.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py37_full/pip_v1.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/nightly_release.sh
View file
Expand Up @@ -33,7 +33,7 @@ export TF_NEED_S3=1
export TF_NEED_CUDA=1
export TF_CUDA_VERSION=10
export TF_CUDNN_VERSION=7
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70
export TF_NEED_TENSORRT=1
export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
export CC_OPT_FLAGS='-mavx'
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/nonpip.sh
View file
Expand Up @@ -36,7 +36,7 @@ export PYTHON_BIN_PATH=$(which python3.8)
export TF2_BEHAVIOR=1
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/release/ubuntu_16/gpu_py38_full/pip.sh
View file
Expand Up @@ -40,7 +40,7 @@ export CC_OPT_FLAGS='-mavx'
export PYTHON_BIN_PATH=$(which ${TF_PYTHON_VERSION})
export PROJECT_NAME="tensorflow_gpu"
export LD_LIBRARY_PATH="/usr/local/cuda:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$TENSORRT_INSTALL_PATH/lib"
export TF_CUDA_COMPUTE_CAPABILITIES=3.5,3.7,5.2,6.0,6.1,7.0
export TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_37,sm_52,sm_60,sm_61,compute_70

yes "" | "$PYTHON_BIN_PATH" configure.py

Expand Down

0 comments on commit cf1b6b3

Please sign in to comment.