(models, models_dir, arch_str)
| 84 | |
| 85 | |
| 86 | def build_model_and_megcc_lib(models, models_dir, arch_str): |
| 87 | # dump all models from onnx to megengine |
| 88 | # models.make(models_dir) |
| 89 | # prepare megcc compiler |
| 90 | prepare_megcc() |
| 91 | # build megcc model lib |
| 92 | for arch_desc in arch_str: |
| 93 | kernel_build_dirs[ |
| 94 | arch_desc] = "{}/benchmark/model/benchmark_kernel_{}".format( |
| 95 | megcc_path, arch_desc) |
| 96 | build_megcc_lib( |
| 97 | arch_desc, |
| 98 | model_config_json="", |
| 99 | kernel_build_dir=kernel_build_dirs[arch_desc], |
| 100 | dtype="fp32", |
| 101 | ) |
| 102 | |
| 103 | |
| 104 | def build_benchmarker(x86_target="fallback", arch_str=None, benchmarkers=None): |
no test coverage detected