()
| 157 | func.get_static_memory_alloc_info(args.get_static_mem_info) |
| 158 | |
| 159 | def run(): |
| 160 | if not args.embed_input: |
| 161 | for key in inp_dict: |
| 162 | inp_dict[key].set_value(mge.Tensor(data[key])._dev_tensor()) |
| 163 | func.execute() |
| 164 | func.wait() |
| 165 | return [oup_node.get_value().numpy() for oup_node in output_dict.values()] |
| 166 | |
| 167 | for i in range(args.warm_up): |
| 168 | logger.info("warming up {}".format(i)) |