feat: add --gpus automatically if tag specified

Author: zapadinsky

src/job.ts

@@ -986,8 +986,8 @@ If you know what you're doing and would like to suppress this warning, use one o
                 dockerCmd += `--cpus=${cpuConfig} `;
             }
 
-            if (this.argv.gpus) {
-                dockerCmd += `--gpus ${this.argv.gpus} --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 `;
+            if (this.gpus) {
+                dockerCmd += `--gpus ${this.gpus} --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 `;
             }
 
             // host and none networks have to be specified using --network, since they cannot be used with
@@ -1223,6 +1223,19 @@ If you know what you're doing and would like to suppress this warning, use one o
         return image.entrypoint;
     }
 
+    get gpus (): string | null {
+        if (this.argv.gpus) {
+            return this.argv.gpus;
+        } else if ("tags" in this.jobData) {
+            for (const tag of this.jobData["tags"]) {
+                if (tag.match(/^(.*-)?gpu(-.*)?$/)) {
+                    return "all";
+                }
+            }
+        }
+        return null;
+    }
+
     private async validateCiDependencyProxyServerAuthentication (imageName: string) {
         const CI_DEPENDENCY_PROXY_SERVER = this._variables["CI_DEPENDENCY_PROXY_SERVER"];
         if (!imageName.startsWith(CI_DEPENDENCY_PROXY_SERVER)) {
@@ -1583,8 +1596,8 @@ If you know what you're doing and would like to suppress this warning, use one o
             dockerCmd += `--shm-size=${this.argv.shmSize} `;
         }
 
-        if (this.argv.gpus) {
-            dockerCmd += `--gpus ${this.argv.gpus} --ipc=host --ulimit memlock=-1 --ulimit stack=67108864`;
+        if (this.gpus) {
+            dockerCmd += `--gpus ${this.gpus} --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 `;
         }
 
         for (const volume of this.argv.volume) {