experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Address potential issues encountered in non-ROCm systems.

Browse Source
This commit is contained in:
Wen-Heng (Jack) Chung 2019-08-02 19:28:52 +00:00
parent 5f3d84ba24
commit 1363c86e2f
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/platform/default/rocm_rocdl_path.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
#include <stdlib.h> #include <stdlib.h>
#if !defined(PLATFORM_GOOGLE) #if !defined(PLATFORM_GOOGLE) && TENSORFLOW_USE_ROCM
#include "third_party/gpus/rocm/rocm_config.h" #include "third_party/gpus/rocm/rocm_config.h"
#endif #endif
#include "tensorflow/core/platform/logging.h" #include "tensorflow/core/platform/logging.h"
@ -25,8 +25,12 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
string ROCmRoot() { string ROCmRoot() {
#if TENSORFLOW_USE_ROCM
VLOG(3) << "ROCM root = " << TF_ROCM_TOOLKIT_PATH; VLOG(3) << "ROCM root = " << TF_ROCM_TOOLKIT_PATH;
return TF_ROCM_TOOLKIT_PATH; return TF_ROCM_TOOLKIT_PATH;
#else
return "";
#endif
} }
} // namespace tensorflow } // namespace tensorflow