Skip to content

Commit 0ee7989

Browse files
authored
Merge pull request #149 from intel/kmb_detection_fix
Fix to detect NPU_3700 and disallow compilation
2 parents 9834be2 + 7d7b10a commit 0ee7989

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

gimpopenvino/plugins/openvino_utils/tools/model_manager.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -320,7 +320,12 @@ def get_npu_architecture(core):
320320
available_devices = core.get_available_devices()
321321
if 'NPU' in available_devices:
322322
architecture = core.get_property('NPU', 'DEVICE_ARCHITECTURE')
323-
return NPU_ARCH_3720 if NPU_ARCH_3720 in architecture else NPU_ARCH_4000
323+
if NPU_ARCH_3720 in architecture:
324+
return NPU_ARCH_3720
325+
elif NPU_ARCH_4000 in architecture:
326+
return NPU_ARCH_4000
327+
return None
328+
324329
except Exception as e:
325330
logging.error(f"Error retrieving NPU architecture: {str(e)}")
326331
return None

0 commit comments

Comments
 (0)