fix nccl compilation to make sure it compiles for architectures that pytorch compiles for (#18704)
Summary:
cc: t-vi gchanan zou3519
This fixes https://github.com/pytorch/pytorch/issues/18359
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18704
Differential Revision: D14717015
Pulled By: soumith
fbshipit-source-id: 4aac036f57e564b05d759662e8ad7a80170901c0
