Skip to content

Commit 610c6f0

Browse files
authored
[PIR] Gen metaconfig in pir grad op (#67835)
1 parent 80c8760 commit 610c6f0

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

paddle/fluid/pir/dialect/op_generator/op_build_gen.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,16 @@
120120
'UnsqueezeInferMeta',
121121
'UnsqueezeWithXShapeInferMeta',
122122
'ArrayPopInferMeta',
123+
# backward.h
124+
'CrossEntropyGradInferMeta',
125+
'CrossEntropyGrad2InferMeta',
126+
'CrossEntropyWithSoftmaxGradInferMeta',
127+
'CSoftmaxWithCrossEntropyGradInferMeta',
128+
'LSTMGradInferMeta',
129+
'FFTC2RGradInferMeta',
130+
'GruGradInferMeta',
131+
'GruUnitGradInferMeta',
132+
'NllLossGradInferMeta',
123133
}
124134

125135
_PREPARE_DATA_WITH_VECTOR_INT64_MTTABLE_ATTRIBUTE = {'FrobeniusNormOp'}

0 commit comments

Comments
 (0)