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

Handle PLATFORM_WINDOWS in stream_executor.

Browse Source 
PiperOrigin-RevId: 265769809
This commit is contained in:
Christian Sigg 2019-08-27 14:38:54 -07:00 committed by TensorFlower Gardener
parent 0e7680ef8d
commit 3367f66a9a
2 changed files with 7 additions and 3 deletions
tensorflow/stream_executor/platform
dso_loader.hplatform.h

2
tensorflow/stream_executor/platform/dso_loader.h
View File

@ -22,7 +22,7 @@ limitations under the License.
#if defined(PLATFORM_GOOGLE)
#include "tensorflow/stream_executor/platform/google/dso_loader.h"
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID)
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_WINDOWS)
#include "tensorflow/stream_executor/platform/default/dso_loader.h"
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform

8
tensorflow/stream_executor/platform/platform.h
View File

@ -16,8 +16,9 @@ limitations under the License.
#ifndef TENSORFLOW_STREAM_EXECUTOR_PLATFORM_PLATFORM_H_
#define TENSORFLOW_STREAM_EXECUTOR_PLATFORM_PLATFORM_H_
#if !defined(PLATFORM_POSIX) && !defined(PLATFORM_GOOGLE) && \
!defined(PLATFORM_POSIX_ANDROID) && !defined(PLATFORM_GOOGLE_ANDROID)
#if !defined(PLATFORM_POSIX) && !defined(PLATFORM_GOOGLE) && \
!defined(PLATFORM_POSIX_ANDROID) && !defined(PLATFORM_GOOGLE_ANDROID) && \
!defined(PLATFORM_WINDOWS)
// Choose which platform we are on.
#if defined(ANDROID) || defined(__ANDROID__)
@ -26,6 +27,9 @@ limitations under the License.
#elif defined(__APPLE__)
#define PLATFORM_POSIX
#elif defined(_WIN32)
#define PLATFORM_WINDOWS
#else
// If no platform specified, use:
#define PLATFORM_POSIX