From 82fac87a038663203b9f820036b0195b1a02456a Mon Sep 17 00:00:00 2001
From: Yuxiang Wei <universe_fly@icloud.com>
Date: Mon, 27 Sep 2021 05:10:12 +0800
Subject: [PATCH] Fix the missing `dtype` attribute of `tir.Shuffle` in Python
 level

---
 include/tvm/tir/expr.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/include/tvm/tir/expr.h b/include/tvm/tir/expr.h
index 8ea48dd592d5..f6741112f269 100644
--- a/include/tvm/tir/expr.h
+++ b/include/tvm/tir/expr.h
@@ -947,6 +947,7 @@ class ShuffleNode : public PrimExprNode {
   Array<PrimExpr> indices;
 
   void VisitAttrs(AttrVisitor* v) {
+    v->Visit("dtype", &dtype);
     v->Visit("vectors", &vectors);
     v->Visit("indices", &indices);
     v->Visit("span", &span);