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

Replace cub-hip with rocprim #78

Closed petrex closed 6 years ago

petrex commented 6 years ago

We need to replace cub-hip with RocM official parallel primitive --> rocPRIM.

The pre-release can be found here:

Code: https://github.com/ROCmSoftwarePlatform/rocPRIM/tree/milestone3_prerelease Release: https://github.com/ROCmSoftwarePlatform/rocPRIM/releases/tag/v0.2.1

oscarbg commented 6 years ago

Hi just exploring this port and found new Project rocPRIM.. seems interestingly replaces cub-hip.. can't find anything under https://github.com/ROCmSoftwarePlatform/rocPRIM? returns error no existant Project.. is for AMD internal use? any ETA on public Access? thanks..

VincentSC commented 6 years ago

@oscarbg It's now public

oscarbg commented 6 years ago

Thanks @VincentSC I downloaded days ago the preview.. still not tested tough on a Vega 56 on the TODO list.. :-(

petrex commented 6 years ago

WIP. #114

VincentSC commented 6 years ago

FYI: we released another dev-branch https://github.com/ROCmSoftwarePlatform/rocPRIM/tree/v0.3.2 Final version v1.0 is expected on Friday.

petrex commented 6 years ago

@VincentSC Do you guys have a plan to make rocprim self-contained?

VincentSC commented 6 years ago

Not sure if I follow you. Can you elaborate?

petrex commented 6 years ago

https://github.com/pytorch/pytorch