Skip to content

Commit 54fc791

Browse files
authored
Fix case bug (#2681)
1 parent 0d80a01 commit 54fc791

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/aie_lit_utils/lit_config_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -258,15 +258,15 @@ def detect_xrt(
258258

259259
# Map model to NPU generation and filter by available components
260260
if model in LitConfigHelper.NPU_MODELS["npu1"]:
261-
if "aie2" in vitis_components:
261+
if "AIE2" in vitis_components:
262262
run_on_npu1 = run_on_npu
263263
config.features.extend(["ryzen_ai", "ryzen_ai_npu1"])
264264
config.substitutions["%run_on_npu1%"] = run_on_npu1
265265
print(f"Running tests on NPU1 with command line: {run_on_npu1}")
266266
else:
267267
print("NPU1 detected but aietools for aie2 not available")
268268
elif model in LitConfigHelper.NPU_MODELS["npu2"]:
269-
if "aie2p" in vitis_components:
269+
if "AIE2P" in vitis_components:
270270
run_on_npu2 = run_on_npu
271271
config.features.extend(["ryzen_ai", "ryzen_ai_npu2"])
272272
config.substitutions["%run_on_npu2%"] = run_on_npu2

0 commit comments

Comments
 (0)