scitas-job-submit/job_submit.lua
48f5ffde77edmaster

job_submit.lua
View Options

			-- cf. slurm/slurm_errno.h
			ESLURM_INVALID_GRES = 2072

			--########################################################################--
			--
			-- Load billing_cost_estimate parameters (can be overriden in rates_file)
			--
			--########################################################################--

			CONF_DIR = '/etc/slurm/bce'
			CONF_FILES = {'cluster.conf', 'rates.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.lua, partition_setting.lua ---
			require(verbose_mode.lua)
			require(track_gres.lua)
			require(scitas_cost.lua)
			require(partition_setting.lua)

			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
			local verbose = 0
			if verbose == 1 then
			verbose_mode(job_desc)
			end

			status = track_gres(job_desc, submit_uid)
			if status ~= 0 then
			return status
			end

			scitas_cost(job_desc, CPU_COST, GPU_COST, CORES_PER_NODE, GPUS_PER_NODE, DEFAULT_WTIME, DEFAULT_PARTITION, PARALLEL_PARTITION, SERIAL_PARTITION, submit_uid)

			return slurm.SUCCESS
			end


			-- The other required function
			function slurm_job_modify(job_desc, job_rec, part_list, modify_uid)
			return slurm.SUCCESS
			end

			slurm.log_info("job_submit_plugin initialized")
			return slurm.SUCCESS