Add missing rocm_config.h include to amdgpu_compiler.cc

rocm_config.h defines `TF_ROCM_VERSION` which is used in `amdgpu_compiler.cc`. PiperOrigin-RevId: 634476794
ROCm · May 17, 2024 · 7488ace · 7488ace
1 parent e4ac4a5
commit 7488ace
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xla/service/gpu/amdgpu_compiler.cc b/xla/service/gpu/amdgpu_compiler.cc
@@ -60,6 +60,10 @@ limitations under the License.
 #include "tsl/platform/statusor.h"
 #include "tsl/platform/threadpool.h"
 
+#if TENSORFLOW_USE_ROCM
+#include "rocm/rocm_config.h"
+#endif
+
 namespace xla {
 namespace gpu {