do not land: try out new gpu worker images

mozilla · Dec 21, 2023 · 36a246c · 36a246c
1 parent d7f725e
commit 36a246c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/taskcluster/config.yml b/taskcluster/config.yml
@@ -34,28 +34,28 @@ workers:
             provisioner: '{trust-domain}-{level}'
             implementation: generic-worker
             os: linux
-            worker-type: 'b-linux-v100-gpu'
+            worker-type: 'b-linux-v100-gpu-relops782-test'
         # Use for tasks that need lots of GPU power, but not lots of disk space
         # eg: translation & scoring
         b-largegpu:
             provisioner: '{trust-domain}-{level}'
             implementation: generic-worker
             os: linux
-            worker-type: 'b-linux-v100-gpu-4'
+            worker-type: 'b-linux-v100-gpu-relops782-test'
         # Use for tasks that needs lots of GPU power and increased disk space
         # eg: bicleaner
         b-largegpu-largedisk:
             provisioner: '{trust-domain}-{level}'
             implementation: generic-worker
             os: linux
-            worker-type: 'b-linux-v100-gpu-4-300gb'
+            worker-type: 'b-linux-v100-gpu-relops782-test'
         # Use for tasks that need lots of GPU power and immensive amounts of disk space
         # eg: training
         b-largegpu-xlargedisk:
             provisioner: '{trust-domain}-{level}'
             implementation: generic-worker
             os: linux
-            worker-type: 'b-linux-v100-gpu-4-1tb'
+            worker-type: 'b-linux-v100-gpu-relops782-test'
         images:
             provisioner: '{trust-domain}-{level}'
             implementation: docker-worker
@@ -65,7 +65,7 @@ workers:
 # Keys are worker type, and align with the `worker-type` entries in the
 # `worker.aliases` above.
 worker-configuration:
-    b-linux-v100-gpu:
+    b-linux-v100-gpu-relops782-test:
         env:
             GPUS: "0"
             WORKSPACE: "12000"

diff --git a/taskcluster/kinds/all-pr/kind.yml b/taskcluster/kinds/all-pr/kind.yml
@@ -20,7 +20,7 @@ kind-dependencies:
 
 tasks:
     all:
-        description: Dummy task that ensures all parts of training pipeline will run
+        description: Dummy task that ensures all parts of training pipeline will run.
         attributes:
             stage: all-pr
             src_locale: "{src_locale}"

diff --git a/taskcluster/translations_taskgraph/target_tasks.py b/taskcluster/translations_taskgraph/target_tasks.py
@@ -20,7 +20,7 @@ def filter(task):
         if task.attributes.get("trg_locale") != trg:
             return False
 
-        # Datasets are only applicable to dataset-specific tasks. If these
+        # Datasets are only applicable to dataset-specific tasks. If these 
         # attribute isn't present on the task it can be assumed to be included
         # if the above attributes matched, as it will be a task that is either
         # agnostic of datasets, or folds in datasets from earlier tasks.