Skip to content

Commit 550cd99

Browse files
authored
Rephrase namespaces for brevity.
Differential Revision: D62189433 Pull Request resolved: #5080
1 parent d32bf99 commit 550cd99

File tree

2 files changed

+90
-122
lines changed

2 files changed

+90
-122
lines changed

extension/module/module.cpp

Lines changed: 36 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -33,42 +33,26 @@
3333
std::move(*et_result__)); \
3434
})
3535

36-
using ::exec_aten::Tensor;
37-
using ::executorch::extension::FileDataLoader;
38-
using ::executorch::extension::MallocMemoryAllocator;
39-
using ::executorch::extension::MmapDataLoader;
40-
using ::executorch::runtime::DataLoader;
41-
using ::executorch::runtime::Error;
42-
using ::executorch::runtime::EValue;
43-
using ::executorch::runtime::EventTracer;
44-
using ::executorch::runtime::HierarchicalAllocator;
45-
using ::executorch::runtime::MemoryAllocator;
46-
using ::executorch::runtime::MemoryManager;
47-
using ::executorch::runtime::MethodMeta;
48-
using ::executorch::runtime::Program;
49-
using ::executorch::runtime::Result;
50-
using ::executorch::runtime::Span;
51-
5236
namespace executorch {
5337
namespace extension {
5438

5539
Module::Module(
5640
const std::string& file_path,
57-
const Module::LoadMode load_mode,
58-
std::unique_ptr<EventTracer> event_tracer)
41+
const LoadMode load_mode,
42+
std::unique_ptr<runtime::EventTracer> event_tracer)
5943
: file_path_(file_path),
6044
load_mode_(load_mode),
6145
memory_allocator_(std::make_unique<MallocMemoryAllocator>()),
6246
temp_allocator_(std::make_unique<MallocMemoryAllocator>()),
6347
event_tracer_(std::move(event_tracer)) {
64-
::executorch::runtime::runtime_init();
48+
runtime::runtime_init();
6549
}
6650

6751
Module::Module(
68-
std::unique_ptr<DataLoader> data_loader,
69-
std::unique_ptr<MemoryAllocator> memory_allocator,
70-
std::unique_ptr<MemoryAllocator> temp_allocator,
71-
std::unique_ptr<EventTracer> event_tracer)
52+
std::unique_ptr<runtime::DataLoader> data_loader,
53+
std::unique_ptr<runtime::MemoryAllocator> memory_allocator,
54+
std::unique_ptr<runtime::MemoryAllocator> temp_allocator,
55+
std::unique_ptr<runtime::EventTracer> event_tracer)
7256
: data_loader_(std::move(data_loader)),
7357
memory_allocator_(
7458
memory_allocator ? std::move(memory_allocator)
@@ -77,14 +61,14 @@ Module::Module(
7761
temp_allocator ? std::move(temp_allocator)
7862
: std::make_unique<MallocMemoryAllocator>()),
7963
event_tracer_(std::move(event_tracer)) {
80-
::executorch::runtime::runtime_init();
64+
runtime::runtime_init();
8165
}
8266

8367
Module::Module(
84-
std::shared_ptr<Program> program,
85-
std::unique_ptr<MemoryAllocator> memory_allocator,
86-
std::unique_ptr<MemoryAllocator> temp_allocator,
87-
std::unique_ptr<EventTracer> event_tracer)
68+
std::shared_ptr<runtime::Program> program,
69+
std::unique_ptr<runtime::MemoryAllocator> memory_allocator,
70+
std::unique_ptr<runtime::MemoryAllocator> temp_allocator,
71+
std::unique_ptr<runtime::EventTracer> event_tracer)
8872
: program_(std::move(program)),
8973
memory_allocator_(
9074
memory_allocator ? std::move(memory_allocator)
@@ -93,10 +77,10 @@ Module::Module(
9377
temp_allocator ? std::move(temp_allocator)
9478
: std::make_unique<MallocMemoryAllocator>()),
9579
event_tracer_(std::move(event_tracer)) {
96-
::executorch::runtime::runtime_init();
80+
runtime::runtime_init();
9781
}
9882

99-
Error Module::load(const Program::Verification verification) {
83+
runtime::Error Module::load(const runtime::Program::Verification verification) {
10084
if (!is_loaded()) {
10185
if (!data_loader_) {
10286
switch (load_mode_) {
@@ -119,15 +103,15 @@ Error Module::load(const Program::Verification verification) {
119103
break;
120104
}
121105
};
122-
auto program =
123-
ET_UNWRAP_UNIQUE(Program::load(data_loader_.get(), verification));
124-
program_ = std::shared_ptr<Program>(
125-
program.release(), [](Program* pointer) { delete pointer; });
106+
auto program = ET_UNWRAP_UNIQUE(
107+
runtime::Program::load(data_loader_.get(), verification));
108+
program_ = std::shared_ptr<runtime::Program>(
109+
program.release(), [](runtime::Program* pointer) { delete pointer; });
126110
}
127-
return Error::Ok;
111+
return runtime::Error::Ok;
128112
}
129113

130-
Result<std::unordered_set<std::string>> Module::method_names() {
114+
runtime::Result<std::unordered_set<std::string>> Module::method_names() {
131115
ET_CHECK_OK_OR_RETURN_ERROR(load());
132116
const auto method_count = program_->num_methods();
133117
std::unordered_set<std::string> result;
@@ -139,7 +123,7 @@ Result<std::unordered_set<std::string>> Module::method_names() {
139123
return result;
140124
}
141125

142-
Error Module::load_method(const std::string& method_name) {
126+
runtime::Error Module::load_method(const std::string& method_name) {
143127
if (!is_method_loaded(method_name)) {
144128
ET_CHECK_OK_OR_RETURN_ERROR(load());
145129

@@ -158,10 +142,11 @@ Error Module::load_method(const std::string& method_name) {
158142
method_holder.planned_spans.emplace_back(
159143
method_holder.planned_buffers.back().data(), buffer_size);
160144
}
161-
method_holder.planned_memory = std::make_unique<HierarchicalAllocator>(Span(
162-
method_holder.planned_spans.data(),
163-
method_holder.planned_spans.size()));
164-
method_holder.memory_manager = std::make_unique<MemoryManager>(
145+
method_holder.planned_memory =
146+
std::make_unique<runtime::HierarchicalAllocator>(runtime::Span(
147+
method_holder.planned_spans.data(),
148+
method_holder.planned_spans.size()));
149+
method_holder.memory_manager = std::make_unique<runtime::MemoryManager>(
165150
memory_allocator_.get(),
166151
method_holder.planned_memory.get(),
167152
temp_allocator_.get());
@@ -171,33 +156,36 @@ Error Module::load_method(const std::string& method_name) {
171156
event_tracer_.get()));
172157
methods_.emplace(method_name, std::move(method_holder));
173158
}
174-
return Error::Ok;
159+
return runtime::Error::Ok;
175160
}
176161

177-
Result<MethodMeta> Module::method_meta(const std::string& method_name) {
162+
runtime::Result<runtime::MethodMeta> Module::method_meta(
163+
const std::string& method_name) {
178164
ET_CHECK_OK_OR_RETURN_ERROR(load_method(method_name));
179165
return methods_.at(method_name).method->method_meta();
180166
}
181167

182-
Result<std::vector<EValue>> Module::execute(
168+
runtime::Result<std::vector<runtime::EValue>> Module::execute(
183169
const std::string& method_name,
184-
const std::vector<EValue>& input) {
170+
const std::vector<runtime::EValue>& input) {
185171
ET_CHECK_OK_OR_RETURN_ERROR(load_method(method_name));
186172
auto& method = methods_.at(method_name).method;
187173

188174
ET_CHECK_OK_OR_RETURN_ERROR(method->set_inputs(
189-
exec_aten::ArrayRef<EValue>(input.data(), input.size())));
175+
exec_aten::ArrayRef<runtime::EValue>(input.data(), input.size())));
190176
ET_CHECK_OK_OR_RETURN_ERROR(method->execute());
191177

192178
const auto outputs_size = method->outputs_size();
193-
std::vector<EValue> outputs(outputs_size);
179+
std::vector<runtime::EValue> outputs(outputs_size);
194180
ET_CHECK_OK_OR_RETURN_ERROR(
195181
method->get_outputs(outputs.data(), outputs_size));
196182

197183
return outputs;
198184
}
199185

200-
Error Module::set_output_data_ptr(EValue output_value, size_t output_index) {
186+
runtime::Error Module::set_output_data_ptr(
187+
runtime::EValue output_value,
188+
size_t output_index) {
201189
ET_CHECK_OK_OR_RETURN_ERROR(load_method("forward"));
202190
auto& output_tensor = output_value.toTensor();
203191
auto& method = methods_.at("forward").method;

0 commit comments

Comments
 (0)