Skip to content

Commit 692fa57

Browse files
[ROCm] Avoid hardcoding soversions (#3102)
[ROCm] Avoid hardcoding soversions for tf-2.20-enhanced Co-authored-by: Dragan Mladjenovic <[email protected]>
1 parent 4962be5 commit 692fa57

File tree

4 files changed

+1062
-14
lines changed

4 files changed

+1062
-14
lines changed

third_party/xla/third_party/gpus/rocm/rocm_config.h.tpl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,5 +24,13 @@ limitations under the License.
2424
#define TF_HIPBLASLT %{hipblaslt_flag}
2525
#define TF_HIPRUNTIME_SOVERSION "%{hip_soversion_number}"
2626
#define TF_ROCBLAS_SOVERSION "%{rocblas_soversion_number}"
27+
#define TF_HIPBLASLT_SOVERSION "%{hipblaslt_soversion_number}"
28+
#define TF_MIOPEN_SOVERSION "%{miopen_soversion_number}"
29+
#define TF_HIPFFT_SOVERSION "%{hipfft_soversion_number}"
30+
#define TF_ROCSOLVER_SOVERSION "%{rocsolver_soversion_number}"
31+
#define TF_HIPSPARSE_SOVERSION "%{hipsparse_soversion_number}"
32+
#define TF_ROCTRACER_SOVERSION "%{roctracer_soversion_number}"
33+
#define TF_HIPSOLVER_SOVERSION "%{hipsolver_soversion_number}"
34+
#define TF_ROCRAND_SOVERSION "%{rocrand_soversion_number}"
2735

2836
#endif // ROCM_ROCM_CONFIG_H_

0 commit comments

Comments
 (0)