deepseek-ai/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM.git synced 2026-11-05 00:47:26 +00:00
Code Issues 162 Projects Packages Wiki Activity

[PR #251] [CLOSED] chore: add __all__ export list for explicit public API #315

New issue
Closed
opened 2026-10-26 21:16:10 +00:00 by bsakaguchi · 0 comments
bsakaguchi commented 2026-10-26 21:16:10 +00:00
Owner
Copy link

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepGEMM/pull/251
Author: @yurekami
Created: 12/25/2025
Status: ❌ Closed

Base: main ← Head: chore/add-all-exports

📝 Commits (1)

  • 61b0319 chore: add all export list for explicit public API

📊 Changes

1 file changed (+49 additions, -0 deletions)

View changed files

📝 deep_gemm/__init__.py (+49 -0)

📄 Description

Summary

Add __all__ to deep_gemm/__init__.py to explicitly define the public API of the package.

Changes

Added a comprehensive __all__ list that includes:

Configuration Functions

  • set_num_sms, get_num_sms
  • set_tc_util, get_tc_util

cuBLASLt Kernels

  • cublaslt_gemm_nt, cublaslt_gemm_nn, cublaslt_gemm_tn, cublaslt_gemm_tt

FP8 GEMM Kernels

  • Normal: fp8_gemm_nt, fp8_gemm_nn, fp8_gemm_tn, fp8_gemm_tt
  • Special: fp8_gemm_nt_skip_head_mid
  • M-Grouped: m_grouped_fp8_gemm_nt_contiguous, m_grouped_fp8_gemm_nn_contiguous, m_grouped_fp8_gemm_nt_masked
  • K-Grouped: k_grouped_fp8_gemm_nt_contiguous, k_grouped_fp8_gemm_tn_contiguous

BF16 GEMM Kernels

  • Normal: bf16_gemm_nt, bf16_gemm_nn, bf16_gemm_tn, bf16_gemm_tt
  • M-Grouped: m_grouped_bf16_gemm_nt_contiguous, m_grouped_bf16_gemm_nn_contiguous, m_grouped_bf16_gemm_nt_masked
  • K-Grouped: k_grouped_bf16_gemm_tn_contiguous

Other Kernels

  • Einsum: einsum, fp8_einsum
  • Attention: fp8_mqa_logits, get_paged_mqa_logits_metadata, fp8_paged_mqa_logits

Utilities

  • transform_sf_into_required_layout, get_mk_alignment_for_contiguous_layout

Submodules

  • testing, utils, legacy

Benefits

  • Better IDE support: Autocompletion shows only public API
  • Clearer documentation: Explicit list of public symbols
  • Proper wildcard imports: from deep_gemm import * works correctly
  • Static analysis: Tools like mypy/pyright understand the public API

🤖 Generated with Claude Code

🔄 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/251 **Author:** [@yurekami](https://github.com/yurekami) **Created:** 12/25/2025 **Status:** ❌ Closed **Base:** `main` ← **Head:** `chore/add-all-exports` --- ### 📝 Commits (1) - [`61b0319`](https://github.com/deepseek-ai/DeepGEMM/commit/61b0319907edc187bd2548db93fe2ea1bff9850b) chore: add __all__ export list for explicit public API ### 📊 Changes **1 file changed** (+49 additions, -0 deletions) <details> <summary>View changed files</summary> 📝 `deep_gemm/__init__.py` (+49 -0) </details> ### 📄 Description ## Summary Add `__all__` to `deep_gemm/__init__.py` to explicitly define the public API of the package. ## Changes Added a comprehensive `__all__` list that includes: ### Configuration Functions - `set_num_sms`, `get_num_sms` - `set_tc_util`, `get_tc_util` ### cuBLASLt Kernels - `cublaslt_gemm_nt`, `cublaslt_gemm_nn`, `cublaslt_gemm_tn`, `cublaslt_gemm_tt` ### FP8 GEMM Kernels - Normal: `fp8_gemm_nt`, `fp8_gemm_nn`, `fp8_gemm_tn`, `fp8_gemm_tt` - Special: `fp8_gemm_nt_skip_head_mid` - M-Grouped: `m_grouped_fp8_gemm_nt_contiguous`, `m_grouped_fp8_gemm_nn_contiguous`, `m_grouped_fp8_gemm_nt_masked` - K-Grouped: `k_grouped_fp8_gemm_nt_contiguous`, `k_grouped_fp8_gemm_tn_contiguous` ### BF16 GEMM Kernels - Normal: `bf16_gemm_nt`, `bf16_gemm_nn`, `bf16_gemm_tn`, `bf16_gemm_tt` - M-Grouped: `m_grouped_bf16_gemm_nt_contiguous`, `m_grouped_bf16_gemm_nn_contiguous`, `m_grouped_bf16_gemm_nt_masked` - K-Grouped: `k_grouped_bf16_gemm_tn_contiguous` ### Other Kernels - Einsum: `einsum`, `fp8_einsum` - Attention: `fp8_mqa_logits`, `get_paged_mqa_logits_metadata`, `fp8_paged_mqa_logits` ### Utilities - `transform_sf_into_required_layout`, `get_mk_alignment_for_contiguous_layout` ### Submodules - `testing`, `utils`, `legacy` ## Benefits - **Better IDE support**: Autocompletion shows only public API - **Clearer documentation**: Explicit list of public symbols - **Proper wildcard imports**: `from deep_gemm import *` works correctly - **Static analysis**: Tools like mypy/pyright understand the public API 🤖 Generated with [Claude Code](https://claude.com/claude-code) --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
bsakaguchi 2026-10-26 21:16:10 +00:00
Sign in to join this conversation.
No Branch/Tag specified
Branches Tags
main
public-sync-20261014
0930-release
nv_dev
fix-mega-moe-fence
public-release-2609
public-release-2607
public-release-2606
mega-update
public-release-260416
ko3n1g/ci/build-wheels
nv_dev_f8e8fb5
nv_dev_950d31a
nv_dev_67fc648
nv_dev_c491439
nv_dev_4ff3f54
v2.1.1.post3
v2.1.1.post2
v2.1.1.post1
v2.1.1
v2.1.0
Labels
Clear labels   
pull-request
Mirrored from GitHub Pull Request

No labels
pull-request
Milestone
Clear milestone
No items
No milestone
Projects
Clear projects
No items
No project
Assignees
Clear assignees
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepGEMM#315
No description provided.