added switch for GPU