[PR #79] [MERGED] add lru-cache for get_best_configs to avoid repeated calculation #234

New issue

Closed

opened 2026-10-26 21:15:10 +00:00 by bsakaguchi · 0 comments

bsakaguchi commented

2026-10-26 21:15:10 +00:00

Owner

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepGEMM/pull/79
Author: @yizhang2077
Created: 4/4/2025
Status: ✅ Merged
Merged: 4/7/2025
Merged by: @LyricZhao

Base: main ← Head: lru-cache-opt

📝 Commits (1)

776bd0c add lru-cache to avoid repeated calculation

📊 Changes

1 file changed (+2 additions, -1 deletions)

View changed files

📝 deep_gemm/jit_kernels/gemm.py (+2 -1)

📄 Description

When using deepgemm, get_best_configs is in the critical path and this function cost is observable. We can use lru-cache to avoid it

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/deepseek-ai/DeepGEMM/pull/79 **Author:** [@yizhang2077](https://github.com/yizhang2077) **Created:** 4/4/2025 **Status:** ✅ Merged **Merged:** 4/7/2025 **Merged by:** [@LyricZhao](https://github.com/LyricZhao) **Base:** `main` ← **Head:** `lru-cache-opt` --- ### 📝 Commits (1) - [`776bd0c`](https://github.com/deepseek-ai/DeepGEMM/commit/776bd0ccccdd319798e9dab85414e1493784da6b) add lru-cache to avoid repeated calculation ### 📊 Changes **1 file changed** (+2 additions, -1 deletions) <details> <summary>View changed files</summary> 📝 `deep_gemm/jit_kernels/gemm.py` (+2 -1) </details> ### 📄 Description When using deepgemm, `get_best_configs` is in the critical path and this function cost is observable. We can use `lru-cache` to avoid it ![image](https://github.com/user-attachments/assets/0163dcb5-61dd-49ba-aa49-8e95c7d686b0) --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

bsakaguchi

2026-10-26 21:15:10 +00:00