message: Fixup since TF internal changed since TF 2.4
diff: @@ -21,8 +21,8 @@ limitations under the License.
  #include "tensorflow/core/common_runtime/gpu/gpu_cudamallocasync_allocator.h"
  +#include "tensorflow/core/common_runtime/device/device_id_utils.h"
  #include "tensorflow/core/common_runtime/gpu/gpu_id.h"
  -#include "tensorflow/core/common_runtime/gpu/gpu_id_utils.h"
  #include "tensorflow/core/common_runtime/gpu/gpu_init.h"
  #include "tensorflow/core/framework/allocator.h"
  #include "tensorflow/core/platform/stream_executor.h"
  @@ -37,7 +37,8 @@ GPUcudaMallocAsyncAllocator::GPUcudaMallocAsyncAllocator(
  name_(absl::StrCat("gpu_async_", platform_gpu_id.value())),
  pool_(nullptr) {
  stream_exec_ =
  - GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie();
  + DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(),
  + platform_gpu_id).ValueOrDie();
  #if CUDA_VERSION < 11020
  LOG(FATAL) << "TF_GPU_ALLOCATOR=cuda_malloc_async need CUDA 11.2 or higher to compile.";
  