Homec4science
Diffusion scitas-job-submit (ncoudene) Browse

scitas-job-submit/job_submit.lua
61df3539d2c2ncoudene

Last Change
Raw File

job_submit.lua
View Options

--########################################################################--
--
-- Load billing_cost_estimate parameters
--
--########################################################################--
CONF_DIR = '/etc/slurm/job_submit'
CONF_FILES = {'job_submit.conf', 'cluster.conf'}
for index, file in ipairs(CONF_FILES) do
filetoload = CONF_DIR.."/"..file
file_fh = io.open(filetoload, "r")
if file_fh == nil then
slurm.log_info("slurm_job_modify: No readable %s found!", filetoload)
else
io.close(file_fh)
dofile(filetoload)
end
end
--- require fonctions verbose_mode, track_gres, scitas_cost, partition_setting ---
f = require('job_submit_fx/functions')
if FX_VERBOSE then
require('job_submit_fx/verbose_mode')
end
if FX_TRACK_GRES then
require('job_submit_fx/track_gres')
end
if FX_SCITAS_COST then
require('job_submit_fx/scitas_cost')
end
if FX_PARTITION then
require('job_submit_fx/partition_setting')
end
if FX_FORCE_ACCOUNT then
require('job_submit_fx/force_account')
end
require('job_submit_fx/validate_job')
function slurm_job_submit(job_desc, part_list, submit_uid)
--- Verbose mode ---
--- To enable verbose mode, you must set the verbose_mode variable to 1
if FX_VERBOSE then
verbose_mode(job_desc)
end
if FX_PARTITION then
local partition = partition_setting (job_desc, submit_uid, INFINITE, SEVENTY, PARALLEL_PARTITION, SERIAL_PARTITION)
if job_desc.partition ~= partition then
job_desc.partition = partition
slurm.log_info("slurm_job_submit: for user %u , setting partition: %s",
submit_uid, partition)
end
end
if FX_TRACK_GRES then
status = track_gres(job_desc, submit_uid)
if status ~= 0 then
return status
end
end
if FX_FORCE_ACCOUNT then
local ret_force_account = force_account(job_desc)
if ret_force_account ~= slurm.SUCCESS then return ret_force_account end
end
-- Job validation
local ret_validate = validate_job(job_desc)
if ret_validate ~= slurm.SUCCESS then return ret_validate end
if FX_SCITAS_COST then
local ret_cost = scitas_cost(job_desc, submit_uid)
if ret_cost ~= slurm.SUCCESS then return ret_cost end
end
return slurm.SUCCESS
end
-- The other required function
-- This function is called by the slurmctld daemon with job modification parameters supplied by the scontrol or sview command.
-- job_rec argugment is a pointer to slurmctld daemon's current data structure for the job to be modified
-- cf. https://slurm.schedmd.com/job_submit_plugins.html
function slurm_job_modify(job_desc, job_rec, part_list, modify_uid)
-- TODO : with job_rec we can access TRES/billing and eventually kill a job if it exceed the capping value
return slurm.SUCCESS
end
slurm.log_info("job_submit_plugin initialized")
return slurm.SUCCESS
c4science · Help