leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

v4.5 tag update (#3202)

Browse Source 
* Python DSL examples reorganization.

* v4.5 tag update.
This commit is contained in:
Junkai-Wu
2026-05-06 08:55:27 +08:00
committed by GitHub
parent f74fea9ce3
commit cb37157db5
351 changed files with 36688 additions and 8117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/cutlass_library/emit_kernel_listing.py
View File

@@ -789,9 +789,11 @@ def emit_gemm_kernel_testlist(manifest, curr_build_dir, arch, mode
}
}
cluster_m_fallback = ctas_per_mma_instruction if dynamic_cluster else cluster_shape_m
cluster_n_fallback = 1 if dynamic_cluster else cluster_shape_n
cluster_k_fallback = 1 if dynamic_cluster else cluster_shape_k
# Fallback cluster shape cannot differ from preferred cluster shape in stream-K kernels.
enable_fallback_cluster = dynamic_cluster and 'stream_k' not in kernel_name
cluster_m_fallback = ctas_per_mma_instruction if enable_fallback_cluster else cluster_shape_m
cluster_n_fallback = 1 if enable_fallback_cluster else cluster_shape_n
cluster_k_fallback = 1 if enable_fallback_cluster else cluster_shape_k
if dynamic_datatype: