[tvm] branch unity updated: [Unity][CUTLASS] Fix for purity tracking (#14891)

[wuwei]

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

wuwei 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 201002e78e [Unity][CUTLASS] Fix for purity tracking (#14891)
201002e78e is described below

commit 201002e78e43d2d5a3580cbfc262e7ff5d63fa39
Author: masahi <masahi...@gmail.com>
AuthorDate: Sat May 20 11:53:34 2023 +0900

    [Unity][CUTLASS] Fix for purity tracking (#14891)
    
    fix for purity
---
 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 212e70b209..f263c9c22f 100644
--- a/python/tvm/contrib/cutlass/build.py
+++ b/python/tvm/contrib/cutlass/build.py
@@ -862,7 +862,7 @@ class CutlassRelaxFunctionAnnotator(relax.PyExprMutator):
     def visit_function_(self, f):
         if "Composite" not in f.attrs:
             body = super().visit_expr(f.body)
-            return relax.Function(f.params, body, f.ret_struct_info, f.attrs, 
f.span)
+            return relax.Function(f.params, body, f.ret_struct_info, 
f.is_pure, f.attrs, f.span)
 
         op_type = f.attrs["Composite"]