Add support for PU domain for GPU/VPU