Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions xla/service/gpu/gpu_executable.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1437,7 +1437,7 @@ absl::Status GpuExecutable::ExecuteThunksWithVaRemapping(
run_options->run_options().command_buffer_va_range_idx();
VaRanges* va_ranges = nullptr;
{
absl::MutexLock lock(&va_ranges_mutex_);
absl::MutexLock lock(va_ranges_mutex_);
auto va_ranges_key = std::make_pair(executor, command_buffer_va_range_idx);
va_ranges = &module_va_ranges_[va_ranges_key];
}
Expand Down Expand Up @@ -1467,7 +1467,7 @@ absl::Status GpuExecutable::ExecuteThunksWithVaRemapping(
// Acquire per-executor mutex to protect VA range operations.
// This ensures only one thread uses the VA ranges at a time for this
// executor.
absl::MutexLock va_lock(&va_ranges->mutex);
absl::MutexLock va_lock(va_ranges->mutex);

// Initialize VA ranges if this is first use (va_reservation is null).
if (va_ranges->va_reservation == nullptr) {
Expand Down
Loading