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

【PaddlePaddle Hackathon No.78】add gather, gather_nd, scatter and scatter_nd op #897

Merged
merged 44 commits into from
Sep 22, 2022
Merged
Changes from 1 commit
Commits
Show all changes
44 commits
Select commit Hold shift + click to select a range
b30a7da
add gather and scatter op
zrr1999 Aug 8, 2022
9675f38
fix bug
zrr1999 Aug 21, 2022
6d02455
fix bug
zrr1999 Aug 22, 2022
0a8872a
fix bug
zrr1999 Aug 22, 2022
dabe72c
fix bug
zrr1999 Aug 23, 2022
3b4e976
fix bug
zrr1999 Aug 24, 2022
cce1792
fix bug
zrr1999 Aug 24, 2022
abca2db
fix bug
zrr1999 Aug 24, 2022
417b389
fix bug
zrr1999 Aug 24, 2022
f3a607c
fix bug
zrr1999 Aug 24, 2022
1a39c01
fix bug
zrr1999 Aug 24, 2022
107c519
fix bug
zrr1999 Aug 24, 2022
bd6cd48
fix bug
zrr1999 Aug 24, 2022
97d9578
fix bug
zrr1999 Aug 24, 2022
4e0c3d5
fix bug
zrr1999 Aug 24, 2022
ba87eca
fix bug
zrr1999 Aug 24, 2022
94834cf
fix bug
zrr1999 Aug 24, 2022
6520697
fix bug
zrr1999 Aug 24, 2022
b5cfedb
fix bug
zrr1999 Aug 24, 2022
8b08876
fix bug
zrr1999 Aug 24, 2022
feaa87b
fix bug
zrr1999 Aug 24, 2022
42c1754
fix bug
zrr1999 Aug 25, 2022
6fde20f
fix bug
zrr1999 Aug 25, 2022
4dc115a
Merge branch 'develop' into gather
zrr1999 Aug 29, 2022
3867a21
fix bug
zrr1999 Aug 29, 2022
a2c4cf4
Merge branch 'develop' into gather
zrr1999 Aug 30, 2022
f6d5ca1
fix bug
zrr1999 Aug 30, 2022
21b3787
Merge branch 'develop' into gather
zrr1999 Sep 12, 2022
3d2e079
fix bugs
zrr1999 Sep 12, 2022
3025621
fix bugs
zrr1999 Sep 14, 2022
7d656a8
Merge branch 'develop' into gather
zrr1999 Sep 14, 2022
c0c97d3
fix bugs
zrr1999 Sep 14, 2022
0c68371
fix bugs
zrr1999 Sep 14, 2022
ad91b7c
fix bugs
zrr1999 Sep 14, 2022
9ad4e2b
add pybind
zrr1999 Sep 15, 2022
658b569
fix bugs
zrr1999 Sep 15, 2022
e0ea015
fix bugs
zrr1999 Sep 15, 2022
8d25e23
fix bugs
zrr1999 Sep 16, 2022
58eb499
fix bugs
zrr1999 Sep 16, 2022
08714e9
add SetRandint function
zrr1999 Sep 18, 2022
f2d5f12
fix bugs
zrr1999 Sep 18, 2022
bdb4ade
fix bugs
zrr1999 Sep 19, 2022
8b1be8e
fix bugs
zrr1999 Sep 19, 2022
671cbe5
fix bugs
zrr1999 Sep 21, 2022
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
Prev Previous commit
Next Next commit
fix bug
zrr1999 committed Aug 24, 2022

Verified

This commit was signed with the committer’s verified signature.
AgeManning Age Manning
commit 65206979089296122878835bb31d8a7a787eaef2
2 changes: 1 addition & 1 deletion cinn/hlir/op/contrib/scatter.cc
Original file line number Diff line number Diff line change
@@ -151,7 +151,7 @@ ir::Tensor ScatterNd(const ir::Tensor &A,
B->shape[B->shape.size() - 2],
indices[pos_axes[i]],
common::AutoSimplify(offset + Expr(i)),
Expr(reinterpret_cast<int>(pos_axes.size()))});
Expr(static_cast<int>(pos_axes.size()))});
if (idx.empty()) {
idx.push_back(cur_idx);
A_indices.push_back(cur_idx);