| 16 | namespace mllm { |
| 17 | |
| 18 | EngineConfigArg engineArgAttach() { |
| 19 | auto& cpu_op_thread = |
| 20 | Argparse::add<int32_t>("-engine_cpu_op_thread|--engine_cpu_op_thread").help("config cpu op thread number"); |
| 21 | auto& dispatch_thread = Argparse::add<int32_t>("-engine_dispatcher_thread|--engine_dispatcher_thread") |
| 22 | .help("config dispatcher pool thread number"); |
| 23 | return { |
| 24 | .cpu_op_thread = cpu_op_thread, |
| 25 | .dispatch_thread = dispatch_thread, |
| 26 | }; |
| 27 | } |
| 28 | |
| 29 | void configEngineWithArgs(const EngineConfigArg& args) { |
| 30 | if (args.cpu_op_thread.isSet()) { |
no test coverage detected