Commit Graph

8 Commits (77b0bf41bee0d701dbaa58764d7ee863d784dbbc)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
zchen0211 2d876b8643 gather scatter fix according to google style
7 years ago
zchen0211 84b8baf196 gather scatter with cuda streams
7 years ago
zchen0211 78808b2091 1 api
7 years ago
zchen0211 b851515b16 merge new op grammar
7 years ago
zchen0211 88a8eedda1 scatter gather gpu
7 years ago
Luo Tao 740c8ba12a remove scatter_op.cu/gather_op.cu as they support only_cpu now
8 years ago
zchen0211 323d4233f3 gather op added with python unittest
8 years ago