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

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

bilalsar wants to merge 1 commit into

ROCm:masterROCm/apex:masterfrom

bilalsar:fix/apex-ut-test-layers-current-devicebilalsar/apex:fix/apex-ut-test-layers-current-deviceCopy head branch name to clipboard