commit f5b4e0e0ea9705d6c37675696365b252b53104a2
author Guangda Lai <laigd@google.com> Fri Apr 26 12:02:57 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Apr 26 13:27:16 2019 -0700
tree 2f5eaba08eb603faf4e30bc2099574b0a9c906e1
parent 2133f7e823f38c4d1b690dc430256dc8337c77e7

Add back action_env TENSORRT_INSTALL_PATH so TF-TRT is built by default in TF gpu builds.

PiperOrigin-RevId: 245462577

configure.py

diff --git a/configure.py b/configure.py
index b4d7778..102481a 100644
--- a/configure.py
+++ b/configure.py
@@ -1452,8 +1452,8 @@
       if validate_cuda_config(environ_cp):
         cuda_env_names = [
             'TF_CUDA_VERSION', 'TF_CUBLAS_VERSION', 'TF_CUDNN_VERSION',
-            'TF_TENSORRT_VERSION', 'TF_NCCL_VERSION', 'TF_CUDA_PATHS',
-            'CUDA_TOOLKIT_PATH'
+            'TF_TENSORRT_VERSION', 'TENSORRT_INSTALL_PATH', 'TF_NCCL_VERSION',
+            'TF_CUDA_PATHS', 'CUDA_TOOLKIT_PATH'
         ]
         for name in cuda_env_names:
           if name in environ_cp: