Skip to content

Commit 4ab9509

Browse files
add A100-80GB executor class
1 parent 9da0d98 commit 4ab9509

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

compute_horde/compute_horde/executor_class.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,12 @@ class ExecutorClassSpec:
3838
gpu_vram_gb=48,
3939
spin_up_time=30, # FIXME: temporary value for debugging
4040
),
41+
ExecutorClass.always_on__gpu_a100_80gb: ExecutorClassSpec(
42+
description="always on, NVIDIA A100 GPU, 80GB variant",
43+
has_gpu=True,
44+
gpu_vram_gb=80,
45+
spin_up_time=10,
46+
),
4147
}
4248

4349
# This is the upper TTL for executors, after which executor pool kills an executor.

compute_horde_sdk/src/compute_horde_core/executor_class.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,4 @@ class ExecutorClass(StrEnum):
1010
spin_up_4min__gpu_24gb = "spin_up-4min.gpu-24gb"
1111
always_on__gpu_24gb = "always_on.gpu-24gb"
1212
always_on__llm__a6000 = "always_on.llm.a6000"
13+
always_on__gpu_a100_80gb = "always_on.gpu.a100-80gb"

0 commit comments

Comments
 (0)