@punks/backend-entity-manager 0.0.293 → 0.0.295

package/dist/cjs/types/platforms/nest/plugins/jobs/aws-batch/manager/types.d.ts

@@ -3,6 +3,7 @@ export type BathJobDefinition = {
     image: string;
     memory: number;
     vcpu: number;
+    gpuMemory?: number;
     defaultCommand: string[];
 };
 export type BatchJobOverrides = {

package/dist/cjs/types/platforms/nest/plugins/jobs/aws-batch/models/index.d.ts

@@ -2,6 +2,7 @@ import { JobDefinition } from "../../../../extensions/jobs/abstractions/definiti
 export type AwsBatchInfrastructureParams = {
     vcpu: number;
     memory: number;
+    gpuMemory?: number;
     dockerImage: string;
 };
 export type AwsBatchInvocationParams = {

package/dist/esm/index.js

@@ -40084,6 +40084,14 @@ let AwsBatchService = AwsBatchService_1 = class AwsBatchService {
                         type: ResourceType.VCPU,
                         value: definition.vcpu.toString(),
                     },
+                    ...(definition.gpuMemory
+                        ? [
+                            {
+                                type: ResourceType.GPU,
+                                value: definition.gpuMemory.toString(),
+                            },
+                        ]
+                        : []),
                 ],
                 networkConfiguration: {
                     assignPublicIp: AssignPublicIp.ENABLED,
@@ -40192,6 +40200,7 @@ let AwsJobsProvider = class AwsJobsProvider {
             image: awsInfraParams.dockerImage,
             jobUid: definition.uid,
             memory: awsInfraParams.memory,
+            gpuMemory: awsInfraParams.gpuMemory,
             vcpu: awsInfraParams.vcpu,
             defaultCommand: awsInvocationParams.startCommand,
         });