v4.3.2 update. (#2839)

2025-12-04 23:14:32 +08:00
parent 52ae719eda
commit bc680c7f67
19 changed files with 169 additions and 301 deletions
--- a/examples/python/CuTeDSL/blackwell/dense_gemm.py
+++ b/examples/python/CuTeDSL/blackwell/dense_gemm.py
@@ -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