DEVICE_COUNT
instead of WORLD_SIZE
to calculate nw
#6324
Merged
DEVICE_COUNT
instead of WORLD_SIZE
to calculate nw
#6324