Newer
Older
/******************************************************************************
* Copyright 2021, Barcelona Supercomputing Center (BSC), Spain
*
* This software was partially supported by the EuroHPC-funded project ADMIRE
* (Project ID: 956748, https://www.admire-eurohpc.eu).
*
* This file is part of scord.
*
* scord is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* scord is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with scord. If not, see <https://www.gnu.org/licenses/>.
*
* SPDX-License-Identifier: GPL-3.0-or-later
*****************************************************************************/
#include <stdlib.h>
#include <stdio.h>
#include <admire.h>
#define NJOB_NODES 50

Alberto Miranda
committed
#define NADHOC_NODES 25
#define NINPUTS 10
#define NOUTPUTS 5
int
main(int argc, char* argv[]) {
fprintf(stderr, "ERROR: no location provided\n");
fprintf(stderr, "Usage: ADM_set_io_resources <SERVER_ADDRESS>\n");
exit(EXIT_FAILURE);
}
int exit_status = EXIT_SUCCESS;
ADM_server_t server = ADM_server_create("tcp", argv[1]);
ADM_node_t* job_nodes = prepare_nodes(NJOB_NODES);
assert(job_nodes);

Alberto Miranda
committed
ADM_node_t* adhoc_nodes = prepare_nodes(NADHOC_NODES);
assert(adhoc_nodes);
ADM_dataset_t* inputs = prepare_datasets("input-dataset-%d", NINPUTS);
assert(inputs);
ADM_dataset_t* outputs = prepare_datasets("output-dataset-%d", NOUTPUTS);
assert(outputs);

Alberto Miranda
committed
ADM_adhoc_resources_t adhoc_resources =
ADM_adhoc_resources_create(adhoc_nodes, NADHOC_NODES);
assert(adhoc_resources);
ADM_adhoc_context_t ctx = ADM_adhoc_context_create(

Alberto Miranda
committed
ADM_ADHOC_MODE_SEPARATE_NEW, ADM_ADHOC_ACCESS_RDWR, adhoc_resources,
100, false);
const char* name = "adhoc_storage_42";
ADM_adhoc_storage_t adhoc_storage;
ADM_return_t ret = ADM_register_adhoc_storage(
server, name, ADM_ADHOC_STORAGE_GEKKOFS, ctx, &adhoc_storage);
if(ret != ADM_SUCCESS) {
"ADM_register_adhoc_storage() remote procedure not completed "
"successfully: %s\n",
ADM_strerror(ret));
exit_status = EXIT_FAILURE;
goto cleanup;
}
ADM_job_resources_t job_resources =
ADM_job_resources_create(job_nodes, NJOB_NODES);
assert(job_resources);
ADM_job_requirements_t reqs = ADM_job_requirements_create(
inputs, NINPUTS, outputs, NOUTPUTS, adhoc_storage);
uint64_t slurm_job_id = 42;
ret = ADM_register_job(server, job_resources, reqs, slurm_job_id, &job);
fprintf(stderr,
"ADM_register_job() remote procedure not completed "
"successfully: %s\n",
ADM_strerror(ret));
exit_status = EXIT_FAILURE;
ADM_adhoc_storage_t tier = NULL;

Alberto Miranda
committed
ADM_adhoc_resources_t resources = NULL;
ret = ADM_set_io_resources(server, job, tier, resources);
if(ret != ADM_SUCCESS) {
fprintf(stderr,
"ADM_set_io_resources() remote procedure not completed "
"successfully: %s\n",
ADM_strerror(ret));
exit_status = EXIT_FAILURE;
goto cleanup;
}
fprintf(stdout, "ADM_set_io_resources() remote procedure completed "
"successfully\n");
cleanup:
ADM_remove_job(server, job);
ADM_server_destroy(server);
exit(exit_status);