search

xuhuisheng / rocm-gfx803

196 stars 9 forks source link

Question: where is the source for tensorflow-rocm? #30

Open refeed opened 1 year ago

refeed commented 1 year ago

Thanks for the helpful guide.

Btw, in the readme.md, to install tensorflow-rocm we need to install the wheel provided from this repo, I assume that there were some changes to the original tensorflow-rocm, could we get the sources that are used for building this wheel? Thanks

xuhuisheng commented 1 year ago

Here is tensorflow of ROCm: https://github.com/rocmsoftwareplatform/tensorflow-upstream

refeed commented 1 year ago

I see, so it is from the official tensorflow-rocm, I assume we can install the tensorflow-rocm in this guide by using pip install tensorflow-rocm==2.8.0 right?

xuhuisheng commented 1 year ago

The official tensorflow-rocm didn't support gfx803 a while.

I rebuild tensorflow-rocm with AMDGPU_TARGETS=gfx803, so we can use tensorflow-rocm on gfx803.

refeed commented 1 year ago

Got it, can I get the script/repo used to rebuild the tensorflow-rocm?

xuhuisheng commented 1 year ago

Something likes: clone git, checkout related branch, change AMDGPU_TARGET to gfx803, execute bazel.

please have a look at these issues: https://github.com/RadeonOpenCompute/ROCm/issues/1880#issuecomment-1399687459 https://github.com/RadeonOpenCompute/ROCm/issues/1880#issuecomment-1543648052