airockchip / rknn-toolkit2

Other
991 stars 104 forks source link

Distinguish which NPU hardware is present #140

Closed koutheir closed 2 months ago

koutheir commented 2 months ago

Is there a way to distinguish which NPU hardware is present on the system? This would allow selectively calling functions that are only supported on some models, e.g., rknn_set_core_mask.

zen-xingle commented 2 months ago

In most firmware, try "cat /proc/device-tree/compatible" to show the soc name. Just like, “rockchip,rk3576-evb1-v10rockchip,rk3576”

koutheir commented 2 months ago

Thank you @zen-xingle! This works on my device, and with some parsing, might even be good enough to identify supported device features.

koutheir commented 2 months ago

Future readers might find this documentation useful.