From 8b5f08320e65914596e4065c0cc8c08e0d815b9e Mon Sep 17 00:00:00 2001 From: LeiWang1999 Date: Fri, 9 Aug 2024 16:33:37 +0000 Subject: [PATCH] always with shared.dyn --- bitblas/base/roller/policy/tensorcore.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitblas/base/roller/policy/tensorcore.py b/bitblas/base/roller/policy/tensorcore.py index 468498fbd..f53090cfc 100644 --- a/bitblas/base/roller/policy/tensorcore.py +++ b/bitblas/base/roller/policy/tensorcore.py @@ -334,7 +334,7 @@ def _score(node, thread): # small is better codegen_dict.shared_scope = "shared.dyn" # Or assume we always use shared memory - # codegen_dict.shared_scope = "shared.dyn" + codegen_dict.shared_scope = "shared.dyn" codegen_dict.complete_config(node) codegen_dict.vectorize = self._plan_vectorize(self.prim_func_node, td, block_size)