Пример #1
0
def create_job(job_uuid, experiment, definition, role=None, resources=None):
    job = ExperimentJob(uuid=job_uuid,
                        experiment=experiment,
                        definition=definition)
    if role:
        job.role = role

    if not resources:
        job.save()
        return

    job_resources = {}
    if resources.memory:
        _resources = resources.memory.to_dict()
        if any(_resources.values()):
            job_resources['memory'] = _resources
    if resources.cpu:
        _resources = resources.cpu.to_dict()
        if any(_resources.values()):
            job_resources['cpu'] = _resources
    if resources.gpu:
        _resources = resources.gpu.to_dict()
        if any(_resources.values()):
            job_resources['gpu'] = _resources
    if job_resources:
        job.resources = JobResources.objects.create(**job_resources)
    job.save()
Пример #2
0
def create_job(job_uuid, experiment, definition, role=None, resources=None):
    job = ExperimentJob(uuid=job_uuid,
                        experiment=experiment,
                        definition=definition)
    if role:
        job.role = role

    if not resources:
        job.save()
        return

    job_resources = {}
    if resources.memory:
        _resources = resources.memory.to_dict()
        if any(_resources.values()):
            job_resources['memory'] = _resources
    if resources.cpu:
        _resources = resources.cpu.to_dict()
        if any(_resources.values()):
            job_resources['cpu'] = _resources
    if resources.gpu:
        _resources = resources.gpu.to_dict()
        if any(_resources.values()):
            job_resources['gpu'] = _resources
    if job_resources:
        job.resources = JobResources.objects.create(**job_resources)
    job.save()
Пример #3
0
def create_job(job_uuid,
               experiment,
               role=None,
               k8s_replica=None,
               sequence=None,
               resources=None,
               node_selector=None,
               affinity=None,
               tolerations=None):
    job = ExperimentJob(uuid=uuid.UUID(job_uuid),
                        experiment=experiment,
                        definition={})

    k8s_replica = k8s_replica or role
    if role:
        job.role = role
    if k8s_replica:
        job.k8s_replica = k8s_replica

    if sequence:
        job.sequence = sequence

    if node_selector:
        job.node_selector = node_selector

    if affinity:
        job.affinity = affinity

    if tolerations:
        job.tolerations = tolerations

    def set_resources():
        job_resources = {}
        if resources.memory:
            _resources = resources.memory.to_dict()
            if any(_resources.values()):
                job_resources['memory'] = _resources
        if resources.cpu:
            _resources = resources.cpu.to_dict()
            if any(_resources.values()):
                job_resources['cpu'] = _resources
        if resources.gpu:
            _resources = resources.gpu.to_dict()
            if any(_resources.values()):
                job_resources['gpu'] = _resources
        if resources.tpu:
            _resources = resources.tpu.to_dict()
            if any(_resources.values()):
                job_resources['tpu'] = _resources
        if job_resources:
            job.resources = JobResources.objects.create(**job_resources)

    if resources:
        set_resources()

    job.save()
Пример #4
0
def create_job(job_uuid,
               experiment,
               definition,
               role=None,
               resources=None,
               node_selector=None,
               affinity=None,
               tolerations=None):
    job = ExperimentJob(uuid=job_uuid,
                        experiment=experiment,
                        definition=definition)
    if role:
        job.role = role

    if node_selector:
        job.node_selector = node_selector

    if affinity:
        job.affinity = affinity

    if tolerations:
        job.tolerations = tolerations

    def set_resources():
        job_resources = {}
        if resources.memory:
            _resources = resources.memory.to_dict()
            if any(_resources.values()):
                job_resources['memory'] = _resources
        if resources.cpu:
            _resources = resources.cpu.to_dict()
            if any(_resources.values()):
                job_resources['cpu'] = _resources
        if resources.gpu:
            _resources = resources.gpu.to_dict()
            if any(_resources.values()):
                job_resources['gpu'] = _resources
        if job_resources:
            job.resources = JobResources.objects.create(**job_resources)

    if resources:
        set_resources()

    job.save()