Skip to content

Commit 2cf707b

Browse files
committed
acclerator type
Signed-off-by: Edward Quarm <equarmjnr@gmail.com>
1 parent ec72fb9 commit 2cf707b

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/model_serving/model_runtime/model_validation/constant.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
AcceleratorType.AMD: "amd.com/gpu",
99
AcceleratorType.GAUDI: "habana.ai/gaudi",
1010
AcceleratorType.SPYRE: Labels.Spyre.SPYRE_COM_GPU,
11-
AcceleratorType.CPU_x86: Labels.CPU_x86.CPU_x86,
11+
AcceleratorType.CPU_x86: Labels.CPU.CPU_x86,
1212
}
1313

1414
TEMPLATE_MAP: dict[str, str] = {

utilities/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ class ROCm:
226226
class Spyre:
227227
SPYRE_COM_GPU: str = "ibm.com/spyre_pf"
228228

229-
class CPU_x86:
229+
class CPU:
230230
CPU_x86: str = "cpu"
231231

232232
class Kueue:

0 commit comments

Comments
 (0)