Fix declaration conflict when hip/math_functions.h is included first

This fixes build failure in TensorFlow 1.8 for HCC
Этот коммит содержится в:
Yaxun Sam Liu committed 2018-08-07 15:44:59 -04:00
1 родитель 56126838cc
Коммит dacb18414e
2 files changed
+142

No files matched your search

+6
Просмотреть файл
@@ -31,6 +31,12 @@ THE SOFTWARE.
#include <limits>
#include <stdint.h>
// HCC's own math functions should be included first, otherwise there will
// be conflicts when hip/math_functions.h is included before hip/hip_runtime.h.
#ifdef __HCC__
#include "kalmar_math.h"
#endif
#pragma push_macro("__DEVICE__")
#pragma push_macro("__RETURN_TYPE")