Add SM90 kernels (#200)

This commit is contained in:
Chenggang Zhao
2025-09-29 17:00:23 +08:00
committed by GitHub
parent 904b721731
commit 80ceeb2c76
10 changed files with 1602 additions and 0 deletions

View File

@@ -41,6 +41,10 @@ from deep_gemm_cpp import (
cublaslt_gemm_tn, cublaslt_gemm_tt,
# Einsum kernels
einsum,
# Attention kernels
fp8_mqa_logits,
get_paged_mqa_logits_metadata,
fp8_paged_mqa_logits,
# Layout kernels
transform_sf_into_required_layout
)