Commit f6497932 authored by 余昆's avatar 余昆
Browse files

Code format 


Former-commit-id: 2f212bb1f02c10d2d48a3386c773671eb7fc052c
parent fa674bbc

core/src/scheduler/SchedInst.cpp

+6 −6
Original line number Diff line number Diff line
@@ -55,8 +55,8 @@ load_simple_config() { 
    // get resources

    auto gpu_ids = get_gpu_pool();



    int32_t build_gpu_id;

    config.GetResourceConfigIndexBuildDevice(build_gpu_id);

    int32_t index_build_device_id;

    config.GetResourceConfigIndexBuildDevice(index_build_device_id);



    // create and connect

    ResMgrInst::GetInstance()->Add(ResourceFactory::Create("disk", "DISK", 0, true, false));
@@ -70,15 +70,15 @@ load_simple_config() { 
    for (auto& gpu_id : gpu_ids) {

        ResMgrInst::GetInstance()->Add(ResourceFactory::Create(std::to_string(gpu_id), "GPU", gpu_id, true, true));

        ResMgrInst::GetInstance()->Connect("cpu", std::to_string(gpu_id), pcie);

        if (build_gpu_id == gpu_id) {

        if (index_build_device_id == gpu_id) {

            find_build_gpu_id = true;

        }

    }



    if (not find_build_gpu_id && build_gpu_id != server::CPU_DEVICE_ID) {

    if (not find_build_gpu_id && index_build_device_id != server::CPU_DEVICE_ID) {

        ResMgrInst::GetInstance()->Add(

            ResourceFactory::Create(std::to_string(build_gpu_id), "GPU", build_gpu_id, true, true));

        ResMgrInst::GetInstance()->Connect("cpu", std::to_string(build_gpu_id), pcie);

            ResourceFactory::Create(std::to_string(index_build_device_id), "GPU", index_build_device_id, true, true));

        ResMgrInst::GetInstance()->Connect("cpu", std::to_string(index_build_device_id), pcie);

    }

}

core/src/scheduler/optimizer/OnlyGPUPass.cpp

+1 −2
Original line number Diff line number Diff line
@@ -46,8 +46,7 @@ OnlyGPUPass::Run(const TaskPtr& task) { 
    auto label = std::make_shared<SpecResLabel>(std::weak_ptr<Resource>(res_ptr));

    task->label() = label;



    ++specified_gpu_id_;

    specified_gpu_id_ = specified_gpu_id_ % gpu_id.size();

    specified_gpu_id_ = (specified_gpu_id_ + 1) % gpu_id.size();

    return true;

}