Skip to content

Commit

Permalink
Revert "Don't run VectorizeLoop pass on custom datatypes"
Browse files Browse the repository at this point in the history
This reverts commit 9bd17934c3914889fed97acd56cf1d13e39c9f42.
We don't need this anymore due to the change made by Logan Weber to add a
SkipVectorize pass (apache#3222).
  • Loading branch information
gussmith23 committed Oct 17, 2019
1 parent d949ed5 commit 1c04c03
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions src/pass/vectorize_loop.cc
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@
#include <unordered_map>
#include <vector>
#include "../arithmetic/compute_expr.h"
#include "../codegen/datatype/registry.h"

namespace tvm {
namespace ir {
Expand Down Expand Up @@ -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_);

Expand Down

0 comments on commit 1c04c03

Please sign in to comment.