aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--tensorflow/contrib/cmake/external/snappy.cmake2
-rw-r--r--tensorflow/core/BUILD2
-rw-r--r--tensorflow/core/platform/posix/port.cc8
-rw-r--r--tensorflow/core/platform/windows/port.cc8
4 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/contrib/cmake/external/snappy.cmake b/tensorflow/contrib/cmake/external/snappy.cmake
index a35d8654fb..2d2451521c 100644
--- a/tensorflow/contrib/cmake/external/snappy.cmake
+++ b/tensorflow/contrib/cmake/external/snappy.cmake
@@ -47,4 +47,4 @@ ExternalProject_Add(snappy
)
# actually enables snappy in the source code
-add_definitions(-DSNAPPY) \ No newline at end of file
+add_definitions(-DTF_USE_SNAPPY)
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD
index 5ab84fec5b..d198a796a7 100644
--- a/tensorflow/core/BUILD
+++ b/tensorflow/core/BUILD
@@ -1410,7 +1410,7 @@ cc_library(
hdrs = LIB_INTERNAL_PUBLIC_HEADERS,
copts = tf_copts(),
defines = tf_additional_lib_defines() + [
- "SNAPPY",
+ "TF_USE_SNAPPY",
] + tf_additional_verbs_lib_defines() +
tf_additional_mpi_lib_defines() +
tf_additional_gdr_lib_defines(),
diff --git a/tensorflow/core/platform/posix/port.cc b/tensorflow/core/platform/posix/port.cc
index 3b17bac808..93a59348c8 100644
--- a/tensorflow/core/platform/posix/port.cc
+++ b/tensorflow/core/platform/posix/port.cc
@@ -29,7 +29,7 @@ limitations under the License.
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
#include "snappy.h"
#endif
#if (defined(__APPLE__) && defined(__MACH__)) || defined(__FreeBSD__)
@@ -126,7 +126,7 @@ void AdjustFilenameForLogging(string* filename) {
}
bool Snappy_Compress(const char* input, size_t length, string* output) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
output->resize(snappy::MaxCompressedLength(length));
size_t outlen;
snappy::RawCompress(input, length, &(*output)[0], &outlen);
@@ -139,7 +139,7 @@ bool Snappy_Compress(const char* input, size_t length, string* output) {
bool Snappy_GetUncompressedLength(const char* input, size_t length,
size_t* result) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
return snappy::GetUncompressedLength(input, length, result);
#else
return false;
@@ -147,7 +147,7 @@ bool Snappy_GetUncompressedLength(const char* input, size_t length,
}
bool Snappy_Uncompress(const char* input, size_t length, char* output) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
return snappy::RawUncompress(input, length, output);
#else
return false;
diff --git a/tensorflow/core/platform/windows/port.cc b/tensorflow/core/platform/windows/port.cc
index 85b53e07c4..e327d53949 100644
--- a/tensorflow/core/platform/windows/port.cc
+++ b/tensorflow/core/platform/windows/port.cc
@@ -20,7 +20,7 @@ limitations under the License.
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
#include "snappy.h"
#endif
@@ -118,7 +118,7 @@ void AdjustFilenameForLogging(string* filename) {
}
bool Snappy_Compress(const char* input, size_t length, string* output) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
output->resize(snappy::MaxCompressedLength(length));
size_t outlen;
snappy::RawCompress(input, length, &(*output)[0], &outlen);
@@ -131,7 +131,7 @@ bool Snappy_Compress(const char* input, size_t length, string* output) {
bool Snappy_GetUncompressedLength(const char* input, size_t length,
size_t* result) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
return snappy::GetUncompressedLength(input, length, result);
#else
return false;
@@ -139,7 +139,7 @@ bool Snappy_GetUncompressedLength(const char* input, size_t length,
}
bool Snappy_Uncompress(const char* input, size_t length, char* output) {
-#ifdef SNAPPY
+#ifdef TF_USE_SNAPPY
return snappy::RawUncompress(input, length, output);
#else
return false;