There is a pr for topk, but it is not sorted. You can use it and keep the sort on CPU for now. It will be faster like that I think.
Le jeu. 29 juin 2017 21:11, Victor Campmany <vcampm...@gmail.com> a écrit : > We are working on both sorts for 1D arrays and sort of an axis of an > nd-array. We are trying to release it as soon as possible. > > El jueves, 29 de junio de 2017, 21:02:08 (UTC-4), Adam Becker escribió: >> >> Has there been any progress? I'm in need of sorted TopK on GPU. I can go >> with CPU sort but seems a bit slow. >> >> On Thursday, June 15, 2017 at 4:01:00 AM UTC+8, Victor Campmany wrote: >>> >>> Hi, >>> >>> We are planning to implement a new GPU accelerated sorting algorithm. >>> We'd like to know which are the most frequent sorting cases that you guys >>> use and the data sizes you are dealing with. For example, sorting a large >>> 1d array, sorting a given axis of a tensor or minibatch, or any other type >>> of sorting you come up with. >>> >> -- > > --- > You received this message because you are subscribed to the Google Groups > "theano-users" group. > To unsubscribe from this group and stop receiving emails from it, send an > email to theano-users+unsubscr...@googlegroups.com. > For more options, visit https://groups.google.com/d/optout. > -- --- You received this message because you are subscribed to the Google Groups "theano-users" group. To unsubscribe from this group and stop receiving emails from it, send an email to theano-users+unsubscr...@googlegroups.com. For more options, visit https://groups.google.com/d/optout.