mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-08 18:18:30 +08:00
86 lines
3.2 KiB
C++
86 lines
3.2 KiB
C++
#include "scheduler/ResourceFactory.h"
|
|
#include "scheduler/ResourceMgr.h"
|
|
#include "scheduler/Scheduler.h"
|
|
#include "scheduler/task/TestTask.h"
|
|
#include "utils/Log.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
|
|
using namespace zilliz::milvus::engine;
|
|
|
|
TEST(normal_test, test1) {
|
|
// ResourceMgr only compose resources, provide unified event
|
|
auto res_mgr = std::make_shared<ResourceMgr>();
|
|
auto disk = res_mgr->Add(ResourceFactory::Create("disk", "ssd"));
|
|
auto cpu = res_mgr->Add(ResourceFactory::Create("cpu"));
|
|
auto gpu1 = res_mgr->Add(ResourceFactory::Create("gpu"));
|
|
auto gpu2 = res_mgr->Add(ResourceFactory::Create("gpu"));
|
|
|
|
auto IO = Connection("IO", 500.0);
|
|
auto PCIE = Connection("IO", 11000.0);
|
|
res_mgr->Connect(disk, cpu, IO);
|
|
res_mgr->Connect(cpu, gpu1, PCIE);
|
|
res_mgr->Connect(cpu, gpu2, PCIE);
|
|
|
|
res_mgr->Start();
|
|
|
|
auto scheduler = new Scheduler(res_mgr);
|
|
scheduler->Start();
|
|
|
|
auto task1 = std::make_shared<TestTask>();
|
|
auto task2 = std::make_shared<TestTask>();
|
|
auto task3 = std::make_shared<TestTask>();
|
|
auto task4 = std::make_shared<TestTask>();
|
|
if (auto observe = disk.lock()) {
|
|
observe->task_table().Put(task1);
|
|
observe->task_table().Put(task2);
|
|
observe->task_table().Put(task3);
|
|
observe->task_table().Put(task4);
|
|
}
|
|
|
|
// if (auto disk_r = disk.lock()) {
|
|
// if (auto cpu_r = cpu.lock()) {
|
|
// if (auto gpu1_r = gpu1.lock()) {
|
|
// if (auto gpu2_r = gpu2.lock()) {
|
|
// std::cout << "<<<<<<<<<<before<<<<<<<<<<" << std::endl;
|
|
// std::cout << "disk:" << std::endl;
|
|
// std::cout << disk_r->task_table().Dump() << std::endl;
|
|
// std::cout << "cpu:" << std::endl;
|
|
// std::cout << cpu_r->task_table().Dump() << std::endl;
|
|
// std::cout << "gpu1:" << std::endl;
|
|
// std::cout << gpu1_r->task_table().Dump() << std::endl;
|
|
// std::cout << "gpu2:" << std::endl;
|
|
// std::cout << gpu2_r->task_table().Dump() << std::endl;
|
|
// std::cout << ">>>>>>>>>>before>>>>>>>>>>" << std::endl;
|
|
// }
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
sleep(5);
|
|
|
|
// if (auto disk_r = disk.lock()) {
|
|
// if (auto cpu_r = cpu.lock()) {
|
|
// if (auto gpu1_r = gpu1.lock()) {
|
|
// if (auto gpu2_r = gpu2.lock()) {
|
|
// std::cout << "<<<<<<<<<<after<<<<<<<<<<" << std::endl;
|
|
// std::cout << "disk:" << std::endl;
|
|
// std::cout << disk_r->task_table().Dump() << std::endl;
|
|
// std::cout << "cpu:" << std::endl;
|
|
// std::cout << cpu_r->task_table().Dump() << std::endl;
|
|
// std::cout << "gpu1:" << std::endl;
|
|
// std::cout << gpu1_r->task_table().Dump() << std::endl;
|
|
// std::cout << "gpu2:" << std::endl;
|
|
// std::cout << gpu2_r->task_table().Dump() << std::endl;
|
|
// std::cout << ">>>>>>>>>>after>>>>>>>>>>" << std::endl;
|
|
// }
|
|
// }
|
|
// }
|
|
// }
|
|
scheduler->Stop();
|
|
res_mgr->Stop();
|
|
|
|
ASSERT_EQ(task1->load_count_, 1);
|
|
ASSERT_EQ(task1->exec_count_, 1);
|
|
}
|