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 setCEName( args ): global ceName ceName = args 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 ) grid = Site.split( '.' )[0] numberOfProcessor = JobParameters.getProcessorFromMJF() if not numberOfProcessor: numberOfProcessor = gConfig.getValue( '/Resources/Sites/%s/%s/CEs/%s/Queues/%s/NumberOfProcessors' % ( grid, Site, ceName, Queue ) ) if not numberOfProcessor: numberOfProcessor = gConfig.getValue( '/Resources/Sites/%s/NumberOfProcessors' % grid ) if not numberOfProcessor: numberOfProcessor = Os.getNumberOfCores() maxRAM = JobParameters.getMemoryFromMJF() if not maxRAM: maxRAM = JobParameters.getMemoryFromProc() gLogger.notice( numberOfProcessor, maxRAM )
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)