Skip to content

Commit d4b80cd

Browse files
authored
feat(config): add moai-accelerator rule (#9)
1 parent 53bf336 commit d4b80cd

2 files changed

Lines changed: 132 additions & 0 deletions

File tree

AGENTS.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,3 +36,4 @@ The commit message should be structured as follows:
3636

3737
- `workflow`: Changes related to CI/CD workflows.
3838
- `deploy`: Changes related to deployment (Helm charts, container files, etc.)
39+
- `config`: Changes related to files hard to manage within helm charts.

config/nfd/moai-accelerator.yaml

Lines changed: 131 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,131 @@
1+
apiVersion: nfd.k8s-sigs.io/v1alpha1
2+
kind: NodeFeatureRule
3+
metadata:
4+
name: moai-accelerator
5+
spec:
6+
# https://github.com/ROCm/gpu-operator/blob/main/helm-charts-k8s/templates/gpu-nfd-default-rule.yaml
7+
# the PCI info is from these websites:
8+
# source1: https://admin.pci-ids.ucw.cz/read/PC/1002
9+
# source2: https://devicehunt.com/view/type/pci/vendor/1002
10+
rules:
11+
# --- AMD Instinct Series ---
12+
- name: "accelerator-amd-mi355x"
13+
labels:
14+
moai.moreh.io/accelerator.vendor: amd
15+
moai.moreh.io/accelerator.model: mi355x
16+
matchAny:
17+
- matchFeatures:
18+
- feature: pci.device
19+
matchExpressions:
20+
vendor: { op: In, value: ["1002"] }
21+
device: { op: In, value: ["75a3"] }
22+
23+
- name: "accelerator-amd-mi350x"
24+
labels:
25+
moai.moreh.io/accelerator.vendor: amd
26+
moai.moreh.io/accelerator.model: mi350x
27+
matchAny:
28+
- matchFeatures:
29+
- feature: pci.device
30+
matchExpressions:
31+
vendor: { op: In, value: ["1002"] }
32+
device: { op: In, value: ["75a0"] }
33+
34+
- name: "accelerator-amd-mi325x"
35+
labels:
36+
moai.moreh.io/accelerator.vendor: amd
37+
moai.moreh.io/accelerator.model: mi325x
38+
matchAny:
39+
- matchFeatures:
40+
- feature: pci.device
41+
matchExpressions:
42+
vendor: { op: In, value: ["1002"] }
43+
device: { op: In, value: ["74a5"] }
44+
45+
- name: "accelerator-amd-mi308x"
46+
labels:
47+
moai.moreh.io/accelerator.vendor: amd
48+
moai.moreh.io/accelerator.model: mi308x
49+
matchAny:
50+
- matchFeatures:
51+
- feature: pci.device
52+
matchExpressions:
53+
vendor: { op: In, value: ["1002"] }
54+
device: { op: In, value: ["74a2"] }
55+
56+
- name: "accelerator-amd-mi300a"
57+
labels:
58+
moai.moreh.io/accelerator.vendor: amd
59+
moai.moreh.io/accelerator.model: mi300a
60+
matchAny:
61+
- matchFeatures:
62+
- feature: pci.device
63+
matchExpressions:
64+
vendor: { op: In, value: ["1002"] }
65+
device: { op: In, value: ["74a0"] }
66+
67+
- name: "accelerator-amd-mi300x"
68+
labels:
69+
moai.moreh.io/accelerator.vendor: amd
70+
moai.moreh.io/accelerator.model: mi300x
71+
matchAny:
72+
- matchFeatures:
73+
- feature: pci.device
74+
matchExpressions:
75+
vendor: { op: In, value: ["1002"] }
76+
device: { op: In, value: ["74a1"] }
77+
78+
- name: "accelerator-amd-mi300x-hf"
79+
labels:
80+
moai.moreh.io/accelerator.vendor: amd
81+
moai.moreh.io/accelerator.model: mi300x-hf
82+
matchAny:
83+
- matchFeatures:
84+
- feature: pci.device
85+
matchExpressions:
86+
vendor: { op: In, value: ["1002"] }
87+
device: { op: In, value: ["74a9"] }
88+
89+
- name: "accelerator-amd-mi250x"
90+
labels:
91+
moai.moreh.io/accelerator.vendor: amd
92+
moai.moreh.io/accelerator.model: mi250x
93+
matchAny:
94+
- matchFeatures:
95+
- feature: pci.device
96+
matchExpressions:
97+
vendor: { op: In, value: ["1002"] }
98+
device: { op: In, value: ["7408"] }
99+
100+
- name: "accelerator-amd-mi250"
101+
labels:
102+
moai.moreh.io/accelerator.vendor: amd
103+
moai.moreh.io/accelerator.model: mi250
104+
matchAny:
105+
- matchFeatures:
106+
- feature: pci.device
107+
matchExpressions:
108+
vendor: { op: In, value: ["1002"] }
109+
device: { op: In, value: ["740c"] }
110+
111+
- name: "accelerator-amd-mi210"
112+
labels:
113+
moai.moreh.io/accelerator.vendor: amd
114+
moai.moreh.io/accelerator.model: mi210
115+
matchAny:
116+
- matchFeatures:
117+
- feature: pci.device
118+
matchExpressions:
119+
vendor: { op: In, value: ["1002"] }
120+
device: { op: In, value: ["740f"] }
121+
122+
- name: "accelerator-amd-mi100"
123+
labels:
124+
moai.moreh.io/accelerator.vendor: amd
125+
moai.moreh.io/accelerator.model: mi100
126+
matchAny:
127+
- matchFeatures:
128+
- feature: pci.device
129+
matchExpressions:
130+
vendor: { op: In, value: ["1002"] }
131+
device: { op: In, value: ["738c", "738e"] }

0 commit comments

Comments
 (0)