Skip to main content
Version: v1.3.0

Spread schedule policy

Allocate metax device using spread schedule policy

To allocate metax device with best performance, you need to only assign metax-tech.com/gpu with annotations hami.io/node-scheduler-policy=spread

apiVersion: v1
kind: Pod
metadata:
name: gpu-pod1
annotations:
hami.io/node-scheduler-policy: "spread" # when this parameter is set to spread, the scheduler will try to find the best topology for this task.
spec:
containers:
- name: ubuntu-container
image: cr.metax-tech.com/public-ai-release/c500/colossalai:2.24.0.5-py38-ubuntu20.04-amd64
imagePullPolicy: IfNotPresent
command: ["sleep","infinity"]
resources:
limits:
metax-tech.com/gpu: 4 # requesting 4 metax GPUs