| /*****************************************************************************\ | |
| * task_cgroup_dmem.h - dmem (device memory) cgroup support for task/cgroup | |
| ***************************************************************************** | |
| * Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. | |
| * | |
| * This file is part of Slurm, a resource management program. | |
| * For details, see <https://slurm.schedmd.com/>. | |
| * Please also read the included file: DISCLAIMER. | |
| * | |
| * Slurm 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 2 of the License, or (at your option) | |
| * any later version. | |
| * | |
| * In addition, as a special exception, the copyright holders give permission | |
| * to link the code of portions of this program with the OpenSSL library under | |
| * certain conditions as described in each individual source file, and | |
| * distribute linked combinations including the two. You must obey the GNU | |
| * General Public License in all respects for all of the code used other than | |
| * OpenSSL. If you modify file(s) with this exception, you may extend this | |
| * exception to your version of the file(s), but you are not obligated to do | |
| * so. If you do not wish to do so, delete this exception statement from your | |
| * version. If you delete this exception statement from all source files in | |
| * the program, then also delete it here. | |
| * | |
| * Slurm 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 Slurm; if not, write to the Free Software Foundation, Inc., | |
| * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
| \*****************************************************************************/ | |
| #ifndef _TASK_CGROUP_DMEM_H_ | |
| #define _TASK_CGROUP_DMEM_H_ | |
| /* initialize dmem subsystem of task/cgroup */ | |
| extern int task_cgroup_dmem_init(void); | |
| /* release dmem subsystem resources */ | |
| extern int task_cgroup_dmem_fini(void); | |
| /* | |
| * Set the device memory limits of the job and step cgroups: for every | |
| * allocated sharing device with a usable dmem region, the allocated shard | |
| * count times the per-shard slice of the device. | |
| */ | |
| extern int task_cgroup_dmem_create(stepd_step_rec_t *step); | |
| #endif |