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} `;
+            if (this.gpus) {
+                dockerCmd += `--gpus ${this.gpus} `;
                 dockerCmd += `--ipc=host `;
                 dockerCmd += `--ulimit memlock=-1 `;
                 dockerCmd += `--ulimit stack=67108864 `;
@@ -1226,6 +1226,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)) {
@@ -1586,8 +1599,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} `;
+        if (this.gpus) {
+            dockerCmd += `--gpus ${this.gpus} `;
             dockerCmd += `--ipc=host `;
             dockerCmd += `--ulimit memlock=-1 `;
             dockerCmd += `--ulimit stack=67108864 `;