【Hackathon 6th Fundable Projects 3 No.243】mul static (#64990)

PaddlePaddle · Jun 13, 2024 · edd8f02 · edd8f02
1 parent c2db102
commit edd8f02
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 252 deletions.
diff --git a/paddle/fluid/operators/mul_op.cc b/paddle/fluid/operators/mul_op.cc
diff --git a/paddle/fluid/operators/ops_signature/mul_sig.cc b/paddle/fluid/operators/ops_signature/mul_sig.cc
diff --git a/paddle/phi/ops/yaml/legacy/static_backward.yaml b/paddle/phi/ops/yaml/legacy/static_backward.yaml
@@ -190,6 +190,16 @@
   kernel :
     func : matmul_triple_grad
 
+- backward_op : matmul_with_flatten_grad
+  forward : matmul_with_flatten (Tensor x, Tensor y, int x_num_col_dims=1, int y_num_col_dims=1) -> Tensor(out)
+  args : (Tensor x, Tensor y, Tensor out_grad, int x_num_col_dims=1, int y_num_col_dims=1)
+  output : Tensor(x_grad), Tensor(y_grad)
+  infer_meta :
+    func : GeneralBinaryGradInferMeta
+    param : [x, y]
+  kernel :
+    func : matmul_with_flatten_grad
+
 - backward_op : max_grad
   forward: max (Tensor x, IntArray axis={0}, bool keepdim=false, bool reduce_all=false, int in_dtype=-1, int out_dtype=-1) -> Tensor(out)
   args : (Tensor x, Tensor out, Tensor out_grad, IntArray axis={}, bool keepdim=false, bool reduce_all=false)

diff --git a/paddle/phi/ops/yaml/legacy/static_ops.yaml b/paddle/phi/ops/yaml/legacy/static_ops.yaml
@@ -405,6 +405,16 @@
     func : matmul
   backward : matmul_grad
 
+- op : matmul_with_flatten
+  args : (Tensor x, Tensor y, int x_num_col_dims = 1, int y_num_col_dims = 1)
+  output : Tensor
+  infer_meta :
+    func : MatmulWithFlattenInferMeta
+  kernel :
+    func : matmul_with_flatten
+    data_type : x
+  backward : matmul_with_flatten_grad
+
 - op : matrix_rank
   args : (Tensor x, Tensor tol_tensor, float tol=0.0f, bool hermitian=false, bool use_default_tol=true)
   output : Tensor(out)

diff --git a/test/cpp/imperative/CMakeLists.txt b/test/cpp/imperative/CMakeLists.txt
@@ -36,7 +36,7 @@ cc_test(
 cc_test(
   test_layer
   SRCS test_layer.cc
-  DEPS layer proto_desc operator op_registry variable_helper mul_op)
+  DEPS layer proto_desc operator op_registry variable_helper generated_op)
 cc_test(
   test_prepare_op
   SRCS test_prepare_op.cc
@@ -57,7 +57,7 @@ cc_test(
        operator
        op_registry
        variable_helper
-       mul_op
+       generated_op
        generated_static_op
        elementwise_add_op)
 cc_test(
@@ -70,12 +70,12 @@ cc_test(
        operator
        op_registry
        variable_helper
-       mul_op
+       generated_op
        elementwise_add_op)
 cc_test(
   test_eager
   SRCS test_eager.cc
-  DEPS tracer layer prepared_operator mul_op)
+  DEPS tracer layer prepared_operator generated_op)
 if(WITH_NCCL
    OR WITH_RCCL
    OR WITH_XPU_BKCL)