diff --git a/projects/clr/hipamd/src/hiprtc/cmake/HIPRTC.cmake b/projects/clr/hipamd/src/hiprtc/cmake/HIPRTC.cmake index ddafe6002d..02a9cc1403 100644 --- a/projects/clr/hipamd/src/hiprtc/cmake/HIPRTC.cmake +++ b/projects/clr/hipamd/src/hiprtc/cmake/HIPRTC.cmake @@ -49,7 +49,10 @@ function(get_hiprtc_macros HIPRTC_DEFINES) #define __launch_bounds__(...) \\\n\ select_impl_(__VA_ARGS__, launch_bounds_impl1, launch_bounds_impl0)(__VA_ARGS__) \n\ #pragma clang diagnostic pop\n\ -#define HIP_INCLUDE_HIP_HIP_RUNTIME_H" +#define HIP_INCLUDE_HIP_HIP_RUNTIME_H\n\ +#define HIP_INCLUDE_HIP_HIP_FP16_H\n\ +#define _HIP_BFLOAT16_H_\n\ +#define HIP_INCLUDE_HIP_HIP_VECTOR_TYPES_H" PARENT_SCOPE) endfunction(get_hiprtc_macros) @@ -70,6 +73,7 @@ macro(generate_hiprtc_header HiprtcHeader) #define INT_MAX __INTMAX_MAX__\n\ #include \"hip/hip_runtime.h\"\n\ #include \"hip/hip_fp16.h\"\n\ +#include \"hip/hip_bfloat16.h\"\n\ #pragma pop_macro(\"CHAR_BIT\")\n\ #pragma pop_macro(\"INT_MAX\")") endmacro(generate_hiprtc_header)