From aaed0483d464ad573d421f8004a4bc42dfb5771d Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Fri, 19 Jun 2020 10:37:46 -0700
Subject: [PATCH] Fix map assign issue in CI test (#5854)

---
 src/tir/transforms/bf16_legalize.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tir/transforms/bf16_legalize.cc b/src/tir/transforms/bf16_legalize.cc
index 07f4775ded50..445ac1cf60cd 100644
--- a/src/tir/transforms/bf16_legalize.cc
+++ b/src/tir/transforms/bf16_legalize.cc
@@ -330,7 +330,7 @@ class BF16LowerRewriter : StmtExprMutator {
       }
     }
     if (buffer_remap.size() != 0) {
-      op->buffer_map.assign(changes.begin(), changes.end());
+      op->buffer_map = Map<Var, Buffer>(changes.begin(), changes.end());
     }
   }
 };