Nvidia (NASDAQ: NVDA) generated more of its revenue in China than it did in the United States. But as the artificial intelligence (AI) boom took off, the U.S. started to become a bigger and bigger source of growth for the chip designer.

AI-diffusion” rules could limit the U.S. chip company’s sales even to friendly countries—and give more opportunity to foreign rivals.

Nvidia is caught in the middle of an escalating trade war between the world’s two largest economies. Nvidia on Tuesday said it will take a $5.5 billion financial hit after Washington placed fresh restrictions on the export of its H20 artificial intelligence chips to China,