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 indices; void VisitAttrs(AttrVisitor* v) { + v->Visit("dtype", &dtype); v->Visit("vectors", &vectors); v->Visit("indices", &indices); v->Visit("span", &span);