提交 112166f4 编写于 作者: L LeGend-AI

use emplace instead of insert

上级 9488de4f
......@@ -15,12 +15,10 @@ TEST(CloneOp, clone_4x3_3_times) {
op_conf.mutable_clone_conf()->set_lbn("clone_test_lbn");
auto clone_op = OpMgr::Singleton().ConstructOp(op_conf);
HashMap<std::string, Shape*> bn2shape_ptr;
std::vector<int64_t> shape_vec = {4, 3};
bn2shape_ptr.insert(
std::make_pair(clone_op->SoleIbn(), new Shape(shape_vec)));
HashMap<std::string, Shape*> bn2shape_ptr{
{clone_op->SoleIbn(), new Shape({4, 3})}};
for (std::string obn : clone_op->output_bns()) {
bn2shape_ptr.insert(std::make_pair(obn, new Shape));
bn2shape_ptr.emplace(obn, new Shape);
}
auto fp = [&bn2shape_ptr](const std::string& bn) {
return bn2shape_ptr.at(bn);
......
......@@ -8,7 +8,6 @@
namespace oneflow {
TEST(InnerProductOp, modelparallel_innerproduct) {
OperatorConf op_conf;
op_conf.set_name("modelparallel_ip_test");
......
......@@ -19,13 +19,13 @@ TEST(MultinomialLogisticLossOp, test_loss_op) {
HashMap<std::string, Shape*> bn2shape_ptr;
std::vector<int64_t> shape_vec = {500, 10};
for (std::string ibn : loss_op->input_bns()) {
bn2shape_ptr.insert(std::make_pair(ibn, new Shape(shape_vec)));
bn2shape_ptr.emplace(ibn, new Shape(shape_vec));
}
for (std::string obn : loss_op->output_bns()) {
bn2shape_ptr.insert(std::make_pair(obn, new Shape));
bn2shape_ptr.emplace(obn, new Shape);
}
for (std::string dtbn : loss_op->data_tmp_bns()) {
bn2shape_ptr.insert(std::make_pair(dtbn, new Shape));
bn2shape_ptr.emplace(dtbn, new Shape);
}
auto fp = [&bn2shape_ptr](const std::string& bn) {
return bn2shape_ptr.at(bn);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册