Fix map assign issue in CI test (#5854)

apache · Jun 19, 2020 · aaed048 · aaed048
1 parent 3c733b8
commit aaed048
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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());
     }
   }
 };