Python get_cluster_status Exemples

Exemple #1

Fichier : job_manager.py Projet : junjieqian/DLWorkspace

def AutoApproveJob(job):
    cluster_status = get_cluster_status()
    jobUser = getAlias(job["userName"])
    jobParams = json.loads(base64.b64decode(job["jobParams"]))
    jobGPU = int(jobParams["resourcegpu"])

    currentGPU = 0
    for user in cluster_status["user_status"]:
        if user["userName"] == jobUser:
            currentGPU = int(user["userGPU"])

    if currentGPU == 0 or currentGPU + jobGPU <= 4:
        ApproveJob(job)

Exemple #2

Fichier : job_manager.py Projet : nautilusshell/QianJiangYuan

def AutoApproveJob(job):

    cluster_status = get_cluster_status()
    jobUser = getAlias(job["userName"])
    jobParams = json.loads(base64.b64decode(job["jobParams"]))
    jobGPU = int(jobParams["resourcegpu"])

    logging.info("start to autoApprove job...")
    currentGPU = 0
    logging.info("currentGPU: " + str(currentGPU) + " jobGPU: " + str(jobGPU))

    for user in cluster_status["user_status"]:
        if user["userName"] == jobUser:
            currentGPU = int(user["userGPU"])

    # Default Auto approval changed to 2GPU
    # if currentGPU == 0 or currentGPU + jobGPU <= 2:
    if currentGPU + jobGPU <= 1:
        ApproveJob(job)