From e0f068cd2e60339ea43ccdf7c20e723f0ec0ae61 Mon Sep 17 00:00:00 2001 From: LeiWang1999 Date: Tue, 6 Aug 2024 17:28:37 +0000 Subject: [PATCH] revert change --- 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 f53090cfc..468498fbd 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)