chn/nixos
1
0
Fork 0
mirror of https://github.com/CHN-beta/nixos.git synced 2026-01-12 04:19:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

packages.sbatch-tui: default manually select GPU

Browse Source
This commit is contained in:
陈浩南
2024-09-08 19:22:26 +08:00
parent 43e9cd48d7
commit 00a45fa97c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
packages/sbatch-tui/src/main.cpp
View File

@@ -17,7 +17,7 @@ int main()
int vasp_version_selected = 0;
std::vector<std::string> vasp_version_entries = { "std", "gam", "ncl" };
int device_type_selected = 0;
std::vector<std::string> device_type_entries = { "any single GPU", "manually select GPU", "CPU" };
std::vector<std::string> device_type_entries = { "manually select GPU", "any single GPU", "CPU" };
std::deque<bool> device_selected = std::deque<bool>(Device.GpuIds.size(), false);
std::vector<std::string> device_entries = Device.GpuIds;
std::string job_name = std::filesystem::current_path().filename().string();
@@ -76,7 +76,7 @@ int main()
devices.push_back(ftxui::Checkbox
(state.device_entries[i], &state.device_selected[i], checkbox_option));
return devices;
}()) | with_separator | ftxui::Maybe([&]{ return state.device_type_selected == 1; }),
}()) | with_separator | ftxui::Maybe([&]{ return state.device_type_selected == 0; }),
ftxui::Container::Vertical
({
ftxui::Input(&state.mpi_threads) | ftxui::size(ftxui::WIDTH, ftxui::GREATER_THAN, 3)
@@ -123,7 +123,7 @@ int main()
{
screen.Loop(request_interface);
if (state.user_command == "quit") return EXIT_FAILURE;
else if (state.device_type_selected == 0)
else if (state.device_type_selected == 1)
state.submit_command =
"sbatch --ntasks=1\n--gpus=1\n--job-name='{}'\n--output='{}'\nvasp-nvidia-{}"_f
(state.job_name, state.output_file, state.vasp_version_entries[state.vasp_version_selected]);