Fix compilation with rocm

rocm-arch / tensorflow-rocm

tensorflow-rocm AUR package

17 stars 12 forks source link

Fix compilation with rocm #67

Closed dkarkoulis closed 8 months ago

dkarkoulis commented 9 months ago

There is a rocm function missing in xla causing linking failure. tensorflow-2.15.0/third_party/xla/xla/stream_executor/rocm/rocm_driver.cc tensorflow-2.15.0/third_party/xla/xla/stream_executor/rocm/rocm_driver_wrapper.h