Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 1 commit into from
Jun 13, 2024
Merged

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
206 changes: 0 additions & 206 deletions paddle/fluid/operators/mul_op.cc
View file
Open in desktop

This file was deleted.

42 changes: 0 additions & 42 deletions paddle/fluid/operators/ops_signature/mul_sig.cc
View file
Open in desktop

This file was deleted.

10 changes: 10 additions & 0 deletions paddle/phi/ops/yaml/legacy/static_backward.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
10 changes: 10 additions & 0 deletions paddle/phi/ops/yaml/legacy/static_ops.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
8 changes: 4 additions & 4 deletions test/cpp/imperative/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -57,7 +57,7 @@ cc_test(
operator
op_registry
variable_helper
mul_op
generated_op
generated_static_op
elementwise_add_op)
cc_test(
Expand All @@ -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)
Expand Down