On Linux I tried to set GPU_MAX_COMPUTE_UNITS environment variable, because CPU_MAX_COMPUTE_UNITS works fine, but it doesn' t work.
GPU_MAX_COMPUTE_UNITS is not a supported flag. Currently there is no way to limit the number of CUs in GPU.