Skip to content

Commit 60ee370

Browse files
author
Zhanwei Wu
committed
Update code for support Linux & update sample code & docs
1 parent 0eedadf commit 60ee370

File tree

9 files changed

+12
-4
lines changed

9 files changed

+12
-4
lines changed

pybind/AppBuilder.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,11 +76,17 @@ PYBIND11_MODULE(appbuilder, m) {
7676
m.attr("__license__") = "BSD-3-Clause";
7777

7878
m.def("model_initialize", &initialize, "Initialize models.");
79+
#ifdef _WIN32
7980
m.def("model_initialize", &initialize_P, "Initialize models.");
81+
#endif
8082
m.def("model_inference", &inference, "Inference models.");
83+
#ifdef _WIN32
8184
m.def("model_inference", &inference_P, "Inference models.");
85+
#endif
8286
m.def("model_destroy", &destroy, "Destroy models.");
87+
#ifdef _WIN32
8388
m.def("model_destroy", &destroy_P, "Destroy models.");
89+
#endif
8490
m.def("memory_create", &create_memory, "Create share memory.");
8591
m.def("memory_delete", &delete_memory, "Delete share memory.");
8692
m.def("set_log_level", &set_log_level, "Set QNN log level.");

samples/python/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,8 @@ python stable_diffusion_v2_1\stable_diffusion_v2_1.py --prompt "spectacular view
5252

5353
| Model | QNN SDK Version | Command |
5454
| ---- | :----: | ---- |
55-
| stable_diffusion_v1_5 | 2.24 | python stable_diffusion_v2_1\stable_diffusion_v2_1.py --prompt "the prompt string ..." |
56-
| stable_diffusion_v2_1 | 2.24 | python stable_diffusion_v1_5\stable_diffusion_v1_5.py --prompt "the prompt string ..." |
55+
| stable_diffusion_v2_1 | 2.24 | python stable_diffusion_v2_1\stable_diffusion_v2_1.py --prompt "the prompt string ..." |
56+
| stable_diffusion_v1_5 | 2.24 | python stable_diffusion_v1_5\stable_diffusion_v1_5.py --prompt "the prompt string ..." |
5757
| riffusion | 2.24 | python riffusion\riffusion.py --prompt "the prompt string ..." |
5858
| real_esrgan_x4plus | 2.28 | python real_esrgan_x4plus\real_esrgan_x4plus.py |
5959
| real_esrgan_general_x4v3 | 2.28 | python real_esrgan_general_x4v3\real_esrgan_general_x4v3.py |
9.35 KB
Loading
-331 KB
Binary file not shown.

samples/python/real_esrgan_general_x4v3/real_esrgan_general_x4v3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def Release():
117117

118118
Init()
119119

120-
Inference(execution_ws + "\\input.png", execution_ws + "\\output.png")
120+
Inference(execution_ws + "\\input.jpg", execution_ws + "\\output.jpg")
121121

122122
Release()
123123

9.35 KB
Loading
-331 KB
Binary file not shown.

samples/python/real_esrgan_x4plus/real_esrgan_x4plus.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def Release():
116116

117117
Init()
118118

119-
Inference(execution_ws + "\\input.png", execution_ws + "\\output.png")
119+
Inference(execution_ws + "\\input.jpg", execution_ws + "\\output.jpg")
120120

121121
Release()
122122

src/Utils/DataUtil.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@
1212
#include <iostream>
1313
#include <numeric>
1414
#include <queue>
15+
#ifdef _WIN32
1516
#include <intrin.h>
17+
#endif
1618
#include "DataUtil.hpp"
1719
#include "Logger.hpp"
1820
#ifndef __hexagon__

0 commit comments

Comments
 (0)