From 02664f154f60a19aa3610b1134aab3c048581e92 Mon Sep 17 00:00:00 2001 From: Adal Chiriliuc Date: Wed, 3 Jul 2024 20:17:05 +0300 Subject: [PATCH] Added 'always_on.gpu-24gb' executor class --- compute_horde/compute_horde/executor_class.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/compute_horde/compute_horde/executor_class.py b/compute_horde/compute_horde/executor_class.py index d62f27cc6..67fbfbee4 100644 --- a/compute_horde/compute_horde/executor_class.py +++ b/compute_horde/compute_horde/executor_class.py @@ -5,8 +5,8 @@ class ExecutorClass(StrEnum): spin_up_4min__gpu_24gb = "spin_up-4min.gpu-24gb" + always_on__gpu_24gb = "always_on.gpu-24gb" # always_on__cpu_16c__ram_64gb = "always_on.cpu-16c.ram-64gb" - # always_on__gpu_24gb = "always_on.gpu-24gb" # always_on__gpu_80gb = "always_on.gpu-80gb" # always_on__gpu_24gb__docker_cached_facilitator = "always_on.gpu-24gb.docker_cached-facilitator" @@ -33,16 +33,17 @@ class ExecutorClassSpec: gpu_vram_gb=24, spin_up_time=int(timedelta(minutes=4).total_seconds()), ), + ExecutorClass.always_on__gpu_24gb: ExecutorClassSpec( + description="always on, generic GPU machine", + has_gpu=True, + gpu_vram_gb=24, + spin_up_time=0, + ), # ExecutorClass.always_on__cpu_16c__ram_64gb: ExecutorClassSpec( # cpu_cores=16, # ram_gb=64, # spin_up_time=0, # ), - # ExecutorClass.always_on__gpu_24gb: ExecutorClassSpec( - # has_gpu=True, - # gpu_vram_gb=24, - # spin_up_time=0, - # ), # ExecutorClass.always_on__gpu_80gb: ExecutorClassSpec( # has_gpu=True, # gpu_vram_gb=80,