search

rocmarchive / realcaffe2

The repo is obsolete. Use at your own risk.
https://github.com/pytorch/pytorch
Apache License 2.0
12 stars 2 forks source link

More ops hipified with cub-hip #65

Closed ashishfarmer closed 6 years ago

petrex commented 6 years ago

looks good. When rocprim is live, Pls cherry-pick the change, replace cub-hip with rocprim and merge back to AMD_develop.