fix(tests): use the rank's own device in test_layers all_gather/reduce_scatter parity - #344

The process '/usr/bin/git' failed with exit code 128

unable to access 'http://git-cache.git-cache.svc.cluster.local/ROCm/apex/': Failed to connect to git-cache.git-cache.svc.cluster.local port 80 after 232 ms: Couldn't connect to server

unable to access 'http://git-cache.git-cache.svc.cluster.local/ROCm/apex/': Failed to connect to git-cache.git-cache.svc.cluster.local port 80 after 224 ms: Couldn't connect to server

unable to access 'http://git-cache.git-cache.svc.cluster.local/ROCm/apex/': Failed to connect to git-cache.git-cache.svc.cluster.local port 80 after 215 ms: Couldn't connect to server