[tvm] branch unity updated: [Unity][CUTLASS] Attention hot fix (#14966)

[ju...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

junrushao pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/unity by this push:
     new 240111fe64 [Unity][CUTLASS] Attention hot fix (#14966)
240111fe64 is described below

commit 240111fe64e2457eb7afa28c72b6d083540ac021
Author: masahi <ma...@gmail.com>
AuthorDate: Sat May 27 01:59:06 2023 +0900

    [Unity][CUTLASS] Attention hot fix (#14966)
    
    Attention hot fix. A bug from #14907
---
 python/tvm/contrib/cutlass/build.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/contrib/cutlass/build.py b/python/tvm/contrib/cutlass/build.py
index b8b65d27bd..1d105bbe82 100644
--- a/python/tvm/contrib/cutlass/build.py
+++ b/python/tvm/contrib/cutlass/build.py
@@ -833,7 +833,7 @@ class CutlassRelaxFunctionAnnotator(relax.PyExprMutator):
 
         if op_attrs.causal_mask is None:
             custom_mask_type = 0
-        if op_attrs.causal_mask == "TopLeft":
+        elif op_attrs.causal_mask == "TopLeft":
             custom_mask_type = 1
         elif op_attrs.causal_mask == "BottomRight":
             custom_mask_type = 2