v4.3.2 update. (#2840)

This commit is contained in:
Junkai-Wu
2025-12-04 23:14:50 +08:00
committed by GitHub
parent f88806b1e3
commit ff35fa561d
19 changed files with 164 additions and 257 deletions

View File

@@ -215,7 +215,7 @@ class DenseGemmKernel:
self.occupancy = 1
self.threads_per_cta = 128
self.smem_capacity = utils.get_smem_capacity_in_bytes("sm_100")
self.smem_capacity = utils.get_smem_capacity_in_bytes()
def _setup_attributes(self):
"""Set up configurations that are dependent on GEMM inputs