Fix __HIP_ARCH_* not defined after including math_functions.h

hcc_detail/math_functions.h used to include hcc_detail/hip_runtime.h.

Removing it has caused regression in TensorFlow 1.8.

Put it back for backward compatibiliity.
Этот коммит содержится в:
Yaxun Sam Liu
2018-08-08 08:55:28 -04:00
родитель 450ea49df0
Коммит b8a0c14171
2 изменённых файлов: 13 добавлений и 0 удалений
+5
Просмотреть файл
@@ -1338,3 +1338,8 @@ __HIP_OVERLOAD2(double, min)
#pragma pop_macro("__HIP_OVERLOAD2")
#pragma pop_macro("__DEVICE__")
#pragma pop_macro("__RETURN_TYPE")
// For backward compatibility.
// There are HIP applications e.g. TensorFlow, expecting __HIP_ARCH_* macros
// defined after including math_functions.h.
#include <hip/hcc_detail/hip_runtime.h>