Enabling NPUs to speed up workload