mirror of
https://github.com/immich-app/immich.git
synced 2025-03-01 15:11:21 +01:00
25 lines
1 KiB
Diff
25 lines
1 KiB
Diff
From c4e6f81b4901b2b8178b964377e8fde7118f0e2e Mon Sep 17 00:00:00 2001
|
|
From: mertalev <101130780+mertalev@users.noreply.github.com>
|
|
Date: Fri, 20 Dec 2024 00:59:21 -0500
|
|
Subject: [PATCH] fix: avoid race condition for rocm conv algo caching
|
|
|
|
---
|
|
onnxruntime/core/providers/rocm/nn/conv.cc | 2 ++
|
|
1 file changed, 2 insertions(+)
|
|
|
|
diff --git a/onnxruntime/core/providers/rocm/nn/conv.cc b/onnxruntime/core/providers/rocm/nn/conv.cc
|
|
index d7f47d07a8..134f8a3b43 100644
|
|
--- a/onnxruntime/core/providers/rocm/nn/conv.cc
|
|
+++ b/onnxruntime/core/providers/rocm/nn/conv.cc
|
|
@@ -122,6 +122,8 @@ Status Conv<T, NHWC>::UpdateState(OpKernelContext* context, bool bias_expected)
|
|
bool input_dims_changed = (s_.last_x_dims != x_dims);
|
|
bool w_dims_changed = (s_.last_w_dims != w_dims);
|
|
if (input_dims_changed || w_dims_changed) {
|
|
+ // lock is needed to avoid race condition during algo search
|
|
+ std::lock_guard<OrtMutex> lock(s_.mutex);
|
|
if (input_dims_changed)
|
|
s_.last_x_dims = gsl::make_span(x_dims);
|
|
|
|
--
|
|
2.43.0
|
|
|