Lightning-AI/lightning-thunder

quantization: process tensors on meta device directly, maybe implement CPU quantization (if it is easy)

Open

#1,111 建立於 2024年9月6日

在 GitHub 查看
 (4 留言) (0 反應) (0 負責人)Python (114 fork)github user discovery
good first issuetransforms

倉庫指標

Star
 (1,460 star)
PR 合併指標
 (PR 指標待抓取)

描述

Currently the BitsAndBytesLinearQuant4bit for submodule always calls bitsandbytes.functional.quantize_4bit. This is somewhat touchy for CPU tensors because quantize_4bit only works on GPU tensors but it is outright not so nice for meta tensors, where we only would need to get the right shapes.

https://github.com/Lightning-AI/lightning-thunder/blob/e64d347def39bb47101efafe4177adf9f77a63ec/thunder/transforms/quantization.py#L93-L103

貢獻者指南

研究方向
實現或修改量化邏輯，以處理元張量而不調用僅適用於GPU的bitsandbytes的quantize 4bit。對於CPU，如果可行，考慮簡單的量化實現。
技術棧
pythonpytorch
領域
backend
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
半天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
大致清晰
前置要求
PythonPyTorch
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
60