Fix map assign issue in CI test (apache#5854)

neo-ai · Jul 2, 2020 · 2684024 · 2684024
1 parent dfecea4
commit 2684024
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());
     }
   }
 };