| 179 | |
| 180 | |
| 181 | def parse_env(dump_dir, bin_dir=None): |
| 182 | script_abs = os.path.abspath(sys.argv[0]) |
| 183 | dev_flag = os.path.exists( |
| 184 | os.path.join(os.path.dirname(script_abs), "release_megcc.sh")) |
| 185 | print("dev flag", dev_flag) |
| 186 | env = {} |
| 187 | if dev_flag: |
| 188 | # dev mode |
| 189 | project_path = os.path.dirname(os.path.dirname(script_abs)) |
| 190 | env["build_script"] = os.path.join(project_path, "runtime", "scripts", |
| 191 | "runtime_build.py") |
| 192 | env["mgb_runner_path"] = find("mgb-runner", os.path.curdir) |
| 193 | env["mgb_to_tinynn_path"] = find("mgb-to-tinynn", os.path.curdir) |
| 194 | env["model_dir"] = dump_dir |
| 195 | env["model_info_dir"] = dump_dir |
| 196 | env["kern_dir"] = dump_dir |
| 197 | env["build_dir"] = os.path.join(dump_dir, "build") |
| 198 | else: |
| 199 | # release mode |
| 200 | env["build_script"] = os.path.join(dump_dir, "runtime/script", |
| 201 | "runtime_build.py") |
| 202 | bin_dir = os.path.abspath(bin_dir) if bin_dir else os.path.dirname( |
| 203 | dump_dir) |
| 204 | env["mgb_runner_path"] = find("mgb-runner", bin_dir) |
| 205 | env["mgb_to_tinynn_path"] = find("mgb-to-tinynn", bin_dir) |
| 206 | env["model_dir"] = os.path.join(dump_dir, "model") |
| 207 | env["kern_dir"] = os.path.join(dump_dir, "kern") |
| 208 | env["build_dir"] = os.path.join(dump_dir, "build") |
| 209 | env["model_info_dir"] = os.path.join(dump_dir, "model_info") |
| 210 | return env |
| 211 | |
| 212 | |
| 213 | def auto_check(model_name_2_all, eps, target_arch, target_host, env, mdl_str, user_input_dir, enable_fp16, enable_aarch32_dot, enable_aarch64_i8mm): |