Skip to content

Commit 95dcf29

Browse files
authored
change (#60844)
1 parent fc27a03 commit 95dcf29

1 file changed

Lines changed: 30 additions & 0 deletions

File tree

  • paddle/fluid/pir/dialect/operator/ir

paddle/fluid/pir/dialect/operator/ir/ops.yaml

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
func : add
2525
inplace : (x -> out)
2626
backward : add_grad
27+
data_transform :
28+
support_trans_dtype : x, y
2729
interfaces : paddle::dialect::InferSymbolicShapeInterface
2830

2931
- op : add_n
@@ -383,6 +385,8 @@
383385
kernel :
384386
func : divide
385387
inplace: (x -> out)
388+
data_transform :
389+
support_trans_dtype : x, y
386390
backward : divide_grad
387391

388392
- op : dropout
@@ -416,6 +420,8 @@
416420
spmd_rule: ElementwiseBinaryInferSpmd
417421
kernel :
418422
func : elementwise_pow
423+
data_transform :
424+
support_trans_dtype : x, y
419425
backward : elementwise_pow_grad
420426

421427
- op : embedding
@@ -474,6 +480,8 @@
474480
spmd_rule: ElementwiseBinaryInferSpmd
475481
kernel :
476482
func : equal
483+
data_transform :
484+
support_trans_dtype : x, y
477485
inplace: (x -> out)
478486

479487
- op : exponential_
@@ -521,6 +529,8 @@
521529
func : ElementwiseInferMeta
522530
kernel :
523531
func : floor_divide
532+
data_transform :
533+
support_trans_dtype : x, y
524534
inplace: (x -> out)
525535

526536
- op : frobenius_norm
@@ -664,6 +674,8 @@
664674
func : CompareInferMeta
665675
kernel :
666676
func : greater_equal
677+
data_transform :
678+
support_trans_dtype : x, y
667679
inplace: (x -> out)
668680

669681
- op : greater_than
@@ -673,6 +685,8 @@
673685
func : CompareInferMeta
674686
kernel :
675687
func : greater_than
688+
data_transform :
689+
support_trans_dtype : x, y
676690
inplace: (x -> out)
677691

678692
- op : hardswish
@@ -712,6 +726,8 @@
712726
func : CompareInferMeta
713727
kernel :
714728
func : less_equal
729+
data_transform :
730+
support_trans_dtype : x, y
715731
inplace: (x -> out)
716732

717733
- op : less_than
@@ -721,6 +737,8 @@
721737
func : CompareInferMeta
722738
kernel :
723739
func : less_than
740+
data_transform :
741+
support_trans_dtype : x, y
724742
inplace: (x -> out)
725743

726744
- op : linspace
@@ -786,6 +804,8 @@
786804
spmd_rule : MatmulInferSpmd
787805
kernel :
788806
func : matmul
807+
data_transform :
808+
support_trans_dtype : x, y
789809
backward : matmul_grad
790810

791811
- op : matrix_rank
@@ -823,6 +843,8 @@
823843
spmd_rule : ElementwiseBinaryInferSpmd
824844
kernel :
825845
func : maximum
846+
data_transform :
847+
support_trans_dtype : x, y
826848
backward : maximum_grad
827849

828850
- op : mean
@@ -879,6 +901,8 @@
879901
func : ElementwiseInferMeta
880902
kernel :
881903
func : minimum
904+
data_transform :
905+
support_trans_dtype : x, y
882906
backward : minimum_grad
883907

884908
- op : mish
@@ -922,6 +946,8 @@
922946
spmd_rule : ElementwiseBinaryInferSpmd
923947
kernel :
924948
func : not_equal
949+
data_transform :
950+
support_trans_dtype : x, y
925951
inplace: (x -> out)
926952

927953
- op : one_hot
@@ -1057,6 +1083,8 @@
10571083
func : ElementwiseInferMeta
10581084
kernel :
10591085
func : remainder
1086+
data_transform :
1087+
support_trans_dtype : x, y
10601088
inplace : (x -> out)
10611089

10621090
- op : repeat_interleave
@@ -1274,6 +1302,8 @@
12741302
kernel :
12751303
func : subtract
12761304
inplace : (x -> out)
1305+
data_transform :
1306+
support_trans_dtype : x, y
12771307
backward : subtract_grad
12781308
interfaces : paddle::dialect::InferSymbolicShapeInterface
12791309

0 commit comments

Comments
 (0)