PR openxla#15267: [ROCm] Fixed compilation issues.

Imported from GitHub PR openxla#15267 Issues were caused by following commits: openxla@d17181b openxla@429da0c Copybara import of the project: -- 66fbafe by Zoran Jovanovic <[email protected]>: [ROCm] Fixed compilation issues. Merging this change closes openxla#15267 COPYBARA_INTEGRATE_REVIEW=openxla#15267 from ROCm:ci_hotfix_20240724 66fbafe PiperOrigin-RevId: 655686460
ROCm · Jul 24, 2024 · 7ec84c5 · 7ec84c5
1 parent a4667a9
commit 7ec84c5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/xla/stream_executor/gpu/BUILD b/xla/stream_executor/gpu/BUILD
@@ -576,8 +576,8 @@ xla_test(
 
 cc_library(
     name = "gpu_cudamallocasync_allocator",
-    srcs = ["gpu_cudamallocasync_allocator.cc"],
-    hdrs = ["gpu_cudamallocasync_allocator.h"],
+    srcs = if_cuda_is_configured(["gpu_cudamallocasync_allocator.cc"]),
+    hdrs = if_cuda_is_configured(["gpu_cudamallocasync_allocator.h"]),
     tags = ["gpu"],
     deps = [
         ":gpu_init_impl",

diff --git a/xla/stream_executor/rocm/rocm_executor.cc b/xla/stream_executor/rocm/rocm_executor.cc
@@ -190,8 +190,7 @@ GpuExecutor::CreateEventBasedTimer(GpuStream* stream, bool use_delay_kernel) {
   TF_ASSIGN_OR_RETURN(auto stop_event, CreateGpuEvent(/*allow_timing=*/true));
   TF_RETURN_IF_ERROR(start_event->Record(stream->gpu_stream()));
   return std::make_unique<GpuTimer>(gpu_context(), std::move(start_event),
-                                    std::move(stop_event), stream,
-                                    std::move(semaphore));
+                                    std::move(stop_event), stream);
 }
 
 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) {