Skip to content

README.md #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

README.md #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 21 additions & 2 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,2 +1,21 @@
# TPU_Gen
This repo contains the framework to generate the TPU and their approximate versions, leveraging the Large Language Models.
# TPU-Gen: LLM-Driven Custom Tensor Processing Unit Generator

📌 **Accepted at ICLAD 2025**

This repository hosts the official implementation, dataset, and experimental results for our paper:

**"TPU-Gen: LLM-Driven Custom Tensor Processing Unit Generator"**
Authored by Kartik Pandit, Deepak Vungarala, Mohammed Elbtity, Shaahin Angizi, Ramtin Zand, and collaborators.

---

## 🧠 TPU-Gen Framework Overview

TPU-Gen is the first end-to-end framework that uses fine-tuned LLMs with RAG (Retrieval-Augmented Generation) to generate both **exact and approximate TPU architectures**.

It consists of:

- ✅ Prompt generation from user input
- ✅ LLMs enhanced with multi-shot learning
- ✅ RAG module for dependency fetching
- ✅ Automated design validation and PPA reporting