projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
3c733b8
)
Fix map assign issue in CI test (#5854)
author
Tianqi Chen
<tqchen@users.noreply.github.com>
Fri, 19 Jun 2020 17:37:46 +0000
(10:37 -0700)
committer
GitHub
<noreply@github.com>
Fri, 19 Jun 2020 17:37:46 +0000
(10:37 -0700)
src/tir/transforms/bf16_legalize.cc
patch
|
blob
|
history
diff --git
a/src/tir/transforms/bf16_legalize.cc
b/src/tir/transforms/bf16_legalize.cc
index
07f4775
..
445ac1c
100644
(file)
--- 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());
}
}
};