From 1c04c03a3d766aab4202205e75b33fe90a09d3bd Mon Sep 17 00:00:00 2001 From: Gus Smith Date: Wed, 22 May 2019 13:59:01 -0700 Subject: [PATCH] Revert "Don't run VectorizeLoop pass on custom datatypes" This reverts commit 9bd17934c3914889fed97acd56cf1d13e39c9f42. We don't need this anymore due to the change made by Logan Weber to add a SkipVectorize pass (#3222). --- src/pass/vectorize_loop.cc | 8 -------- 1 file changed, 8 deletions(-) diff --git a/src/pass/vectorize_loop.cc b/src/pass/vectorize_loop.cc index a0735a3c6c50..aed92d692e53 100644 --- a/src/pass/vectorize_loop.cc +++ b/src/pass/vectorize_loop.cc @@ -29,7 +29,6 @@ #include #include #include "../arithmetic/compute_expr.h" -#include "../codegen/datatype/registry.h" namespace tvm { namespace ir { @@ -103,13 +102,6 @@ class Vectorizer : public IRMutator { // user mutate from parent. using IRMutator::Mutate; - // Don't run on custom datatypes. - Expr Mutate(Expr expr) final { - if (tvm::datatype::Registry::Global()->GetTypeRegistered(expr.type().code())) - return expr; - return IRMutator::Mutate(expr); - } - Stmt Mutate(Stmt stmt) final { CHECK(!need_scalarize_);