Skip to content

Commit b5bc405

Browse files
authored
fix elementwise_add (PaddlePaddle#76346)
* fix elementwise_add * fix
1 parent c8dbf0d commit b5bc405

5 files changed

+7
-7
lines changed

paddle/fluid/framework/ir/embedding_eltwise_layernorm_fuse_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -477,4 +477,4 @@ REGISTER_PASS_CAPABILITY(embedding_eltwise_layernorm_fuse_pass)
477477
paddle::framework::compatible::OpVersionComparatorCombination()
478478
.LE("lookup_table", 1)
479479
.LE("lookup_table_v2", 1)
480-
.LE("elementweise_add", 1));
480+
.LE("elementwise_add", 1));

paddle/fluid/framework/ir/preln_embedding_eltwise_layernorm_fuse_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -458,4 +458,4 @@ REGISTER_PASS_CAPABILITY(preln_embedding_eltwise_layernorm_fuse_pass)
458458
paddle::framework::compatible::OpVersionComparatorCombination()
459459
.LE("lookup_table", 1)
460460
.LE("lookup_table_v2", 1)
461-
.LE("elementweise_add", 1));
461+
.LE("elementwise_add", 1));

paddle/fluid/framework/ir/preln_embedding_eltwise_layernorm_fuse_pass.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,8 @@ struct PrelnSkipLayerNorm : public PatternBase {
124124
// (word, weights_0) lookup_table -> word_emb
125125
// (pos, weights_1) lookup_table -> pos_emb
126126
// (sent, weights_2) lookup_table -> sent_emb
127-
// (word_emb, pos_emb) elementweise_add -> elementwise_out_0
128-
// (elemtwise_out_0, sent_emb) elementweise_add -> elementwise_out_1
127+
// (word_emb, pos_emb) elementwise_add -> elementwise_out_0
128+
// (elemtwise_out_0, sent_emb) elementwise_add -> elementwise_out_1
129129
// (elementwise_out_1, scale, bias) layer_norm -> layer_norm_out
130130
//
131131
// and then convert the corresponding subgraph to:

paddle/fluid/framework/ir/trt_embedding_eltwise_layernorm_fuse_pass.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,8 @@ struct TrtSkipLayerNorm : public PatternBase {
124124
// (word, weights_0) lookup_table -> word_emb
125125
// (pos, weights_1) lookup_table -> pos_emb
126126
// (sent, weights_2) lookup_table -> sent_emb
127-
// (word_emb, pos_emb) elementweise_add -> elementwise_out_0
128-
// (elemtwise_out_0, sent_emb) elementweise_add -> elementwise_out_1
127+
// (word_emb, pos_emb) elementwise_add -> elementwise_out_0
128+
// (elemtwise_out_0, sent_emb) elementwise_add -> elementwise_out_1
129129
// (elementwise_out_1, scale, bias) layer_norm -> layer_norm_out
130130
//
131131
// and then convert the corresponding subgraph to:

paddle/fluid/framework/ir/trt_prompt_tuning_embedding_eltwise_layernorm_fuse_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -585,4 +585,4 @@ REGISTER_PASS_CAPABILITY(
585585
paddle::framework::compatible::OpVersionComparatorCombination()
586586
.LE("lookup_table", 1)
587587
.LE("lookup_table_v2", 1)
588-
.LE("elementweise_add", 1));
588+
.LE("elementwise_add", 1));

0 commit comments

Comments
 (0)