From e5a127478c6ed4a649d51a68143114a12a5b8e5d Mon Sep 17 00:00:00 2001
From: senlyu163 <senlyu@163.com>
Date: Thu, 1 Aug 2024 21:20:10 +0800
Subject: [PATCH] [Bugfix][Cutlass] fix cutlass attention template

---
 python/tvm/contrib/cutlass/attention_operation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/contrib/cutlass/attention_operation.py b/python/tvm/contrib/cutlass/attention_operation.py
index 518778ec52ed..69298453cb87 100644
--- a/python/tvm/contrib/cutlass/attention_operation.py
+++ b/python/tvm/contrib/cutlass/attention_operation.py
@@ -111,7 +111,7 @@ def instantiate_attention_template(attrs):
     if (accumulator_buf_size <= ${workspace}->shape[0]) {
         p.output_accum_ptr = static_cast<float*>(${workspace}->data);
     } else {
-        accumulator_buf_size = true;
+        accumulator_buf_allocated = true;
         cudaMalloc(
           &p.output_accum_ptr,
           accumulator_buf_size