kpm: add kaui support

Signed-off-by: Pierre-Alexandre Meyer <pierre@mouraf.org>