Commit Graph

8 Commits (9894a4fb35fc91646474611864cd933a89fd4a9b)

Author SHA1 Message Date
zhaoyuchen2018 b16274556a
Add dscending for argsort (#21400)
5 years ago
zhaoyuchen2018 08c19c585d
Improve argsort performance. (#21267)
5 years ago
Tao Luo 02270b3eb1
remove unused assert.h (#19529)
6 years ago
Yibing Liu 9386ac0a40 Enhance cuda code & unittest for argsort_op
7 years ago
Yibing Liu 92cfa2be3a Avoid using dynamic array in cuda kernel
7 years ago
Yibing Liu 94e72ea6e7 Support more negative axes in argsort_op
7 years ago
Yibing Liu 42645ff779 Compute target index on gpu
7 years ago
Yibing Liu 6ee22c4f71 Add gpu kernel for argsort op
7 years ago