Started implementing multi-GPU support