US immigration policy may be unnecessarily pushing away talent but the assumption that talented Chinese researchers would naturally remain in America unless prevented from doing so ignores the growth of Chinese universities/labs, companies, their funding, national prestige etc.
I mean, don't get me wrong, US is still highly attractive, it is just no longer the only place where an ambitious Chinese researcher can do important work and grow.