| 70 | class CompileScope { |
| 71 | public: |
| 72 | CompileScope(const of::JobConfigProto& job_config, const of::Device& device) { |
| 73 | of::JobConfigProto mut_job_config = job_config; |
| 74 | const std::shared_ptr<of::Scope> scope = CHECK_JUST(MakeScope(mut_job_config, device)); |
| 75 | CHECK_JUST(of::ThreadLocalScopeStackPush(scope)); |
| 76 | |
| 77 | CHECK_JUST(of::JobBuildAndInferCtx_Open(mut_job_config.job_name())); |
| 78 | CHECK_JUST(CHECK_JUST(of::GetCurInferCtx())->SetJobConf(mut_job_config)); |
| 79 | } |
| 80 | |
| 81 | ~CompileScope() { |
| 82 | CHECK_JUST(of::JobBuildAndInferCtx_Close()); |
nothing calls this directly
no test coverage detected