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()
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()
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()
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()