Skip to content
Snippets Groups Projects
Commit 1fbca041 authored by Erwan Rouchet's avatar Erwan Rouchet Committed by Valentin Rigal
Browse files

Always enable WorkerRun.use_gpu on versions requiring GPUs

parent 1920af58
No related branches found
No related tags found
1 merge request!2409Always enable WorkerRun.use_gpu on versions requiring GPUs
......@@ -8,9 +8,11 @@ from arkindex.process.models import FeatureUsage
def migrate_gpu_usage(apps, schema):
WorkerRun = apps.get_model("process", "WorkerRun")
WorkerRun.objects.filter(
version__gpu_usage__in=(FeatureUsage.Required, FeatureUsage.Supported), process__use_gpu=True
).update(use_gpu=True)
# For versions that require GPUs, WorkerRuns should always use a GPU.
WorkerRun.objects.filter(version__gpu_usage=FeatureUsage.Required).update(use_gpu=True)
# For versions that support GPUs, only use a GPU on processes where GPU usage was enabled.
# This replicates the previous behavior with process-wide GPU usage.
WorkerRun.objects.filter(version__gpu_usage=FeatureUsage.Supported, process__use_gpu=True).update(use_gpu=True)
def reverse(apps, schema):
Process = apps.get_model("process", "Process")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment