Added multi-gpu support