Fix __HIP_DEVICE_COMPILE__ not defined when hip/math_functions.h is included

This fixes build failure in TensorFlow 1.8 for HCC
Dieser Commit ist enthalten in:
Yaxun Sam Liu committed 2018-08-07 17:14:39 -04:00
1 Ursprung 69bbf45b44
Commit 058eb7535b
2 Dateien geändert
+15

Keine Dateien entsprechen Ihrer Suche

+1
Datei anzeigen
@@ -27,6 +27,7 @@ THE SOFTWARE.
// paths to provide a consistent include env and avoid "missing symbol" errors that only appears
// on NVCC path:
#include <hip/hip_common.h>
#if defined(__HIP_PLATFORM_HCC__) && !defined(__HIP_PLATFORM_NVCC__)
#include <hip/hcc_detail/math_functions.h>