Skip to content

Commit d978911

Browse files
authored
[onert/python] Separate prepare from session constructor (#14525)
This commit separates prepare function from initializing session instance. ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>
1 parent 80e94f4 commit d978911

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

runtime/onert/api/python/include/nnfw_api_wrapper.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ class NNFW_SESSION
131131

132132
void close_session();
133133
void set_input_tensorinfo(uint32_t index, const tensorinfo *tensor_info);
134+
void prepare();
134135
void run();
135136
void run_async();
136137
void wait();

runtime/onert/api/python/src/nnfw_api_wrapper.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,6 @@ NNFW_SESSION::NNFW_SESSION(const char *package_file_path, const char *backends)
173173
ensure_status(nnfw_create_session(&(this->session)));
174174
ensure_status(nnfw_load_model_from_file(this->session, package_file_path));
175175
ensure_status(nnfw_set_available_backends(this->session, backends));
176-
ensure_status(nnfw_prepare(this->session));
177176
}
178177
NNFW_SESSION::~NNFW_SESSION()
179178
{
@@ -199,6 +198,7 @@ void NNFW_SESSION::set_input_tensorinfo(uint32_t index, const tensorinfo *tensor
199198
}
200199
ensure_status(nnfw_set_input_tensorinfo(session, index, &ti));
201200
}
201+
void NNFW_SESSION::prepare() { ensure_status(nnfw_prepare(session)); }
202202
void NNFW_SESSION::run() { ensure_status(nnfw_run(session)); }
203203
void NNFW_SESSION::run_async() { ensure_status(nnfw_run_async(session)); }
204204
void NNFW_SESSION::wait() { ensure_status(nnfw_await(session)); }

0 commit comments

Comments
 (0)