From ba3a6b5be8662ac1a6d02a5f68240a332ac4d718 Mon Sep 17 00:00:00 2001 From: Federico Negri Date: Mon, 8 Apr 2024 10:34:35 +0200 Subject: [PATCH 1/2] expose node local storage requirement --- src/ansys/hps/client/jms/schema/task_definition.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/ansys/hps/client/jms/schema/task_definition.py b/src/ansys/hps/client/jms/schema/task_definition.py index bf1755eb2..506874e87 100644 --- a/src/ansys/hps/client/jms/schema/task_definition.py +++ b/src/ansys/hps/client/jms/schema/task_definition.py @@ -54,6 +54,10 @@ class Meta: queue = fields.Str( allow_none=True, metadata={"description": "Name of the job scheduler queue."} ) + use_local_scratch = fields.Bool( + allow_none=True, + metadata={"description": "Whether to use node local storage."}, + ) class ResourceRequirementsSchema(BaseSchema): From 71460eb384e879ed0f838c58c93b0c45455d12a8 Mon Sep 17 00:00:00 2001 From: Federico Negri Date: Mon, 8 Apr 2024 10:40:16 +0200 Subject: [PATCH 2/2] update resources --- src/ansys/hps/client/jms/resource/task_definition.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/ansys/hps/client/jms/resource/task_definition.py b/src/ansys/hps/client/jms/resource/task_definition.py index a1b0f2a10..cc7ca8a74 100644 --- a/src/ansys/hps/client/jms/resource/task_definition.py +++ b/src/ansys/hps/client/jms/resource/task_definition.py @@ -52,6 +52,8 @@ class HpcResources(Object): Whether a job can't share resources with other running jobs. queue : str, optional Name of the job scheduler queue. + use_local_scratch : bool, optional + Whether to use node local storage. """ class Meta: @@ -64,12 +66,14 @@ def __init__( num_gpus_per_node: int = missing, exclusive: bool = missing, queue: str = missing, + use_local_scratch: bool = missing, **kwargs, ): self.num_cores_per_node = num_cores_per_node self.num_gpus_per_node = num_gpus_per_node self.exclusive = exclusive self.queue = queue + self.use_local_scratch = use_local_scratch self.obj_type = self.__class__.__name__