Good things come in small packages: Should we adopt Lite-GPUs in AI infrastructure?

التفاصيل البيبلوغرافية
العنوان: Good things come in small packages: Should we adopt Lite-GPUs in AI infrastructure?
المؤلفون: Canakci, Burcu, Liu, Junyi, Wu, Xingbo, Cheriere, Nathanaël, Costa, Paolo, Legtchenko, Sergey, Narayanan, Dushyanth, Rowstron, Ant
سنة النشر: 2025
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Hardware Architecture, Computer Science - Artificial Intelligence, Computer Science - Distributed, Parallel, and Cluster Computing
الوصف: To match the blooming demand of generative AI workloads, GPU designers have so far been trying to pack more and more compute and memory into single complex and expensive packages. However, there is growing uncertainty about the scalability of individual GPUs and thus AI clusters, as state-of-the-art GPUs are already displaying packaging, yield, and cooling limitations. We propose to rethink the design and scaling of AI clusters through efficiently-connected large clusters of Lite-GPUs, GPUs with single, small dies and a fraction of the capabilities of larger GPUs. We think recent advances in co-packaged optics can be key in overcoming the communication challenges of distributing AI workloads onto more Lite-GPUs. In this paper, we present the key benefits of Lite-GPUs on manufacturing cost, blast radius, yield, and power efficiency; and discuss systems opportunities and challenges around resource, workload, memory, and network management.
Comment: 5+ pages, 4 figures
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2501.10187
رقم الانضمام: edsarx.2501.10187
قاعدة البيانات: arXiv