gpu: boost: fix logic for AK