Skip to content
This repository has been archived by the owner on Jan 24, 2024. It is now read-only.

add PEs(matmul, all relus) and Ops(elementwise_add, relu6, mul) and C… #198

Merged
merged 4 commits into from
Sep 2, 2020

Conversation

wenming2014
Copy link
Collaborator

add PEs(matmul, all relus) and Ops(elementwise_add, relu6, mul) and C++/python tests(#138 )

Copy link
Collaborator

@haozech haozech left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

cinn/hlir/framework/print_graph_pass_test.cc Outdated Show resolved Hide resolved
cinn/hlir/op/nn.cc Show resolved Hide resolved
cinn/hlir/op/nn.cc Show resolved Hide resolved
cinn/hlir/op/op_broadcast_test.cc Outdated Show resolved Hide resolved
cinn/hlir/pe/broadcast.cc Outdated Show resolved Hide resolved
cinn/hlir/pe/transform.cc Outdated Show resolved Hide resolved
cinn/hlir/pe/transform.cc Outdated Show resolved Hide resolved
cinn/hlir/pe/transform.cc Outdated Show resolved Hide resolved
cinn/hlir/pe/transform.h Show resolved Hide resolved
python/tests/test_op_broadcast.py Outdated Show resolved Hide resolved
@wenming2014 wenming2014 force-pushed the develop branch 2 times, most recently from a1d30ed to e46f501 Compare September 2, 2020 03:27
cinn/hlir/pe/pe_broadcast_test.cc Show resolved Hide resolved
*shape1_new = shape1;
*shape2_new = shape2;
if (trans_a) {
reverse(shape1_new->begin(), shape1_new->end());
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

reverse 应该需要 std:: 前缀吧,STL 统一用 std:: 吧

Copy link
Collaborator

@Superjomn Superjomn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

cinn/hlir/pe/pe_elementwise_test.cc Outdated Show resolved Hide resolved
@wenming2014 wenming2014 merged commit 9b6e04c into PaddlePaddle:develop Sep 2, 2020
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants