diff options
Diffstat (limited to 'tensorflow/core/util/port.cc')
-rw-r--r-- | tensorflow/core/util/port.cc | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/core/util/port.cc b/tensorflow/core/util/port.cc new file mode 100644 index 0000000000..12eb076a4d --- /dev/null +++ b/tensorflow/core/util/port.cc @@ -0,0 +1,13 @@ +#include "tensorflow/core/util/port.h" + +namespace tensorflow { + +bool IsGoogleCudaEnabled() { +#if GOOGLE_CUDA + return true; +#else + return false; +#endif +} + +} // end namespace tensorflow |