pyxis/args.h at main · NVIDIA/pyxis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/*
 * Copyright (c) 2020-2022, NVIDIA CORPORATION. All rights reserved.
 */

#ifndef ARGS_H_
#define ARGS_H_

#include <stdbool.h>
#include <stddef.h>

#include <slurm/spank.h>

struct plugin_args {
	char *image;
	char **mounts;
	size_t mounts_len;
	char *workdir;
	char *container_name;
	char *container_name_flags;
	char *container_save;
	int mount_home;
	int remap_root;
	int entrypoint;
	int entrypoint_log;
	int writable;
	char **env_vars;
	size_t env_vars_len;
};

struct plugin_args *pyxis_args_register(spank_t sp);

bool pyxis_args_enabled(void);

int add_mount(const char *source, const char *target, const char *flags);

void remove_all_mounts(void);

void pyxis_args_free(void);

void pyxis_args_check_environment_variables(spank_t sp);

#endif /* ARGS_H_ */