Skip to content

Commit 065a0f0

Browse files
committed
refactor: rename generate_mutex_ to inference_mutex_
1 parent 3d19723 commit 065a0f0

File tree

6 files changed

+9
-9
lines changed

6 files changed

+9
-9
lines changed

packages/react-native-executorch/common/rnexecutorch/models/embeddings/text/TextEmbeddings.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,13 @@ TokenIdsWithAttentionMask TextEmbeddings::preprocess(const std::string &input) {
3636
}
3737

3838
void TextEmbeddings::unload() noexcept {
39-
std::scoped_lock lock(generate_mutex_);
39+
std::scoped_lock lock(inference_mutex_);
4040
BaseModel::unload();
4141
}
4242

4343
std::shared_ptr<OwningArrayBuffer>
4444
TextEmbeddings::generate(const std::string input) {
45-
std::scoped_lock lock(generate_mutex_);
45+
std::scoped_lock lock(inference_mutex_);
4646
auto preprocessed = preprocess(input);
4747

4848
std::vector<int32_t> tokenIdsShape = {

packages/react-native-executorch/common/rnexecutorch/models/embeddings/text/TextEmbeddings.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class TextEmbeddings final : public BaseEmbeddings {
2424
void unload() noexcept;
2525

2626
private:
27-
mutable std::mutex generate_mutex_;
27+
mutable std::mutex inference_mutex_;
2828
std::vector<std::vector<int32_t>> inputShapes;
2929
TokenIdsWithAttentionMask preprocess(const std::string &input);
3030
std::unique_ptr<TokenizerModule> tokenizer;

packages/react-native-executorch/common/rnexecutorch/models/text_to_image/TextToImage.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ std::shared_ptr<OwningArrayBuffer>
5858
TextToImage::generate(std::string input, int32_t imageSize,
5959
size_t numInferenceSteps, int32_t seed,
6060
std::shared_ptr<jsi::Function> callback) {
61-
std::scoped_lock lock(generate_mutex_);
61+
std::scoped_lock lock(inference_mutex_);
6262
setImageSize(imageSize);
6363
setSeed(seed);
6464

@@ -138,7 +138,7 @@ size_t TextToImage::getMemoryLowerBound() const noexcept {
138138
}
139139

140140
void TextToImage::unload() noexcept {
141-
std::scoped_lock lock(generate_mutex_);
141+
std::scoped_lock lock(inference_mutex_);
142142
encoder->unload();
143143
unet->unload();
144144
decoder->unload();

packages/react-native-executorch/common/rnexecutorch/models/text_to_image/TextToImage.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ class TextToImage final {
5050
static constexpr float guidanceScale = 7.5f;
5151
static constexpr float latentsScale = 0.18215f;
5252
bool interrupted = false;
53-
mutable std::mutex generate_mutex_;
53+
mutable std::mutex inference_mutex_;
5454

5555
std::shared_ptr<react::CallInvoker> callInvoker;
5656
std::unique_ptr<Scheduler> scheduler;

packages/react-native-executorch/common/rnexecutorch/models/voice_activity_detection/VoiceActivityDetection.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,13 @@ VoiceActivityDetection::preprocess(std::span<float> waveform) const {
5555
}
5656

5757
void VoiceActivityDetection::unload() noexcept {
58-
std::scoped_lock lock(generate_mutex_);
58+
std::scoped_lock lock(inference_mutex_);
5959
BaseModel::unload();
6060
}
6161

6262
std::vector<types::Segment>
6363
VoiceActivityDetection::generate(std::span<float> waveform) const {
64-
std::scoped_lock lock(generate_mutex_);
64+
std::scoped_lock lock(inference_mutex_);
6565

6666
auto windowedInput = preprocess(waveform);
6767
auto [chunksNumber, remainder] = std::div(

packages/react-native-executorch/common/rnexecutorch/models/voice_activity_detection/VoiceActivityDetection.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ class VoiceActivityDetection : public BaseModel {
2626
void unload() noexcept;
2727

2828
private:
29-
mutable std::mutex generate_mutex_;
29+
mutable std::mutex inference_mutex_;
3030
std::vector<std::array<float, constants::kPaddedWindowSize>>
3131
preprocess(std::span<float> waveform) const;
3232
std::vector<types::Segment> postprocess(const std::vector<float> &scores,

0 commit comments

Comments
 (0)