PR #10607: [ROCM] fixing missing includes and API change
PR #10607: [ROCM] fixing missing includes and API change

Imported from GitHub PR https://github.com/openxla/xla/pull/10607

This is a recent build-brake fix due to API changes

@xla-rotation: would you please have a look ?
Copybara import of the project:

--
ad9a9626fb22d05e137f8b8bad9d81efacda352f by Pavel Emeliyanenko <pavel.emeliyanenko@amd.com>:

fixing missing includes and API change

Merging this change closes #10607

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10607 from ROCm:ci_build_brake_fix_240315 ad9a9626fb22d05e137f8b8bad9d81efacda352f
