def main():
global ceName
global Site
global Queue
Script.registerSwitch("N:", "Name=", "Computing Element Name (Mandatory)",
setCEName)
Script.registerSwitch("S:", "Site=", "Site Name (Mandatory)", setSite)
Script.registerSwitch("Q:", "Queue=", "Queue Name (Mandatory)", setQueue)
Script.parseCommandLine(ignoreErrors=True)
gLogger.info("Getting number of processors")
numberOfProcessor = JobParameters.getNumberOfProcessors(
Site, ceName, Queue)
gLogger.info("Getting memory (RAM) from MJF")
maxRAM = JobParameters.getMemoryFromMJF()
if not maxRAM:
gLogger.info(
"maxRAM could not be found in MJF, using JobParameters.getMemoryFromProc()"
)
maxRAM = JobParameters.getMemoryFromProc()
gLogger.info("Getting number of GPUs")
numberOfGPUs = JobParameters.getNumberOfGPUs(Site, ceName, Queue)
# just communicating it back
gLogger.notice(" ".join(
str(wnPar) for wnPar in [numberOfProcessor, maxRAM, numberOfGPUs]))
def setSite(args):
global Site
Site = args
def setQueue(args):
global Queue
Queue = args
Script.registerSwitch("N:", "Name=", "Computing Element Name (Mandatory)",
setCEName)
Script.registerSwitch("S:", "Site=", "Site Name (Mandatory)", setSite)
Script.registerSwitch("Q:", "Queue=", "Queue Name (Mandatory)", setQueue)
Script.parseCommandLine(ignoreErrors=True)
gLogger.info("Getting number of processors")
numberOfProcessor = JobParameters.getNumberOfProcessors(Site, ceName, Queue)
gLogger.info("Getting memory (RAM) from MJF")
maxRAM = JobParameters.getMemoryFromMJF()
if not maxRAM:
gLogger.info(
"maxRAM could not be found in MJF, using JobParameters.getMemoryFromProc()"
)
maxRAM = JobParameters.getMemoryFromProc()
# just communicating it back
gLogger.notice(numberOfProcessor, maxRAM)