We need to extend the code to Mutiple cpu , multiple gpus as available in most seamless way possible without making things convoluted.