From 3f6ba1020dadc34a8f83cab1938b65c6e230d03b Mon Sep 17 00:00:00 2001
From: Philippe Tillet <ptillet@g.harvard.edu>
Date: Sun, 28 Mar 2021 17:08:38 -0400
Subject: [PATCH] [CODEGEN] Make sure peephole is called before anything else
 in codegen

---
 lib/runtime/function.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/runtime/function.cc b/lib/runtime/function.cc
index 91d126de5..1087571a8 100644
--- a/lib/runtime/function.cc
+++ b/lib/runtime/function.cc
@@ -147,6 +147,8 @@ std::tuple<std::shared_ptr<driver::module>,
   codegen::generator isel(&axes, &layouts, &align, &allocation, &swizzle, target.get(), opt.num_warps);
   // run passes
   dce.run(ir);
+  peephole.run(ir);
+  dce.run(ir);
   pipeline.run(ir);
   dce.run(ir);
   disassociate.run(ir);