Skip to content

Commit b73312f

Browse files
authored
Pass images by value to allow rvalue args.
Differential Revision: D62141582 Pull Request resolved: #5034
1 parent f824c1e commit b73312f

File tree

6 files changed

+8
-5
lines changed

6 files changed

+8
-5
lines changed

examples/models/llava/main.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,6 @@ int32_t main(int32_t argc, char** argv) {
103103
.width = static_cast<int32_t>(image_tensor.size(2)),
104104
.height = static_cast<int32_t>(image_tensor.size(1))}};
105105
// generate
106-
runner.generate(images, prompt, seq_len);
106+
runner.generate(std::move(images), prompt, seq_len);
107107
return 0;
108108
}

examples/models/llava/runner/llava_runner.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ Error LlavaRunner::load() {
7474
}
7575

7676
Error LlavaRunner::generate(
77-
std::vector<Image>& images,
77+
std::vector<Image> images,
7878
const std::string& prompt,
7979
int32_t seq_len,
8080
std::function<void(const std::string&)> token_callback,

examples/models/llava/runner/llava_runner.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class LlavaRunner : public MultimodalRunner {
3131
bool is_loaded();
3232
Error load();
3333
Error generate(
34-
std::vector<Image>& images,
34+
std::vector<Image> images,
3535
const std::string& prompt,
3636
int32_t seq_len = 1024,
3737
std::function<void(const std::string&)> token_callback = {},

examples/models/llava/runner/targets.bzl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@ def define_common_targets():
88
visibility = [
99
"@EXECUTORCH_CLIENTS",
1010
],
11+
compiler_flags = [
12+
"-Wno-global-constructors",
13+
],
1114
exported_deps = [
1215
"//executorch/backends/xnnpack:xnnpack_backend",
1316
"//executorch/extension/llm/runner:runner_lib",

extension/android/jni/jni_layer_llama.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ class ExecuTorchLlamaJni
165165
images.push_back(image_runner);
166166
}
167167
multi_modal_runner_->generate(
168-
images,
168+
std::move(images),
169169
prompt->toStdString(),
170170
seq_len,
171171
[callback](std::string result) { callback->onResult(result); },

extension/llm/runner/multimodal_runner.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ class MultimodalRunner {
5656
virtual bool is_loaded() = 0;
5757
virtual ::executorch::runtime::Error load() = 0;
5858
virtual ::executorch::runtime::Error generate(
59-
std::vector<Image>& images,
59+
std::vector<Image> images,
6060
const std::string& prompt,
6161
int32_t seq_len = 1024,
6262
std::function<void(const std::string&)> token_callback = {},

0 commit comments

Comments
 (0)