File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments