Skip to content

Conversation

@copybara-service
Copy link

PR #27498: [ROCm] Add new hip_runtime bazel target

Imported from GitHub PR #27498

In the jaxlib extension for ROCm, we need to link to the HIP runtime (libamdhip64.so) but without the extra links that are on rocm_hip, such as comgr, rocm-smi, and HSA runtime (which are linked by libamdhip itself)

Comgr shouldn't be directly linked by any HIP users either, but that hasn't been an issue for others apparently, so add this target that we can use from jaxlib and not intefere with rocm_hip.
Copybara import of the project:

--
82a3ab3 by Mathew Odden [email protected]:

[ROCm] Add new hip_runtime bazel target

In the jaxlib extension for ROCm, we need to link to
the HIP runtime (libamdhip64.so) but without the extra
links that are on rocm_hip, such as comgr, rocm-smi, and
HSA runtime (which are linked by libamdhip itself)

Comgr shouldn't be directly linked by any HIP users either,
but that hasn't been an issue for others apparently, so
add this target that we can use from jaxlib and not intefere
with rocm_hip.

Merging this change closes #27498

FUTURE_COPYBARA_INTEGRATE_REVIEW=#27498 from mrodden:rocm-add-hip-runtime-target 82a3ab3

Imported from GitHub PR #27498

In the jaxlib extension for ROCm, we need to link to the HIP runtime (libamdhip64.so) but without the extra links that are on rocm_hip, such as comgr, rocm-smi, and HSA runtime (which are linked by libamdhip itself)

Comgr shouldn't be directly linked by any HIP users either, but that hasn't been an issue for others apparently, so add this target that we can use from jaxlib and not intefere with rocm_hip.
Copybara import of the project:

--
82a3ab3 by Mathew Odden <[email protected]>:

[ROCm] Add new hip_runtime bazel target

In the jaxlib extension for ROCm, we need to link to
the HIP runtime (libamdhip64.so) but without the extra
links that are on rocm_hip, such as comgr, rocm-smi, and
HSA runtime (which are linked by libamdhip itself)

Comgr shouldn't be directly linked by any HIP users either,
but that hasn't been an issue for others apparently, so
add this target that we can use from jaxlib and not intefere
with rocm_hip.

Merging this change closes #27498

COPYBARA_INTEGRATE_REVIEW=#27498 from mrodden:rocm-add-hip-runtime-target 82a3ab3
PiperOrigin-RevId: 770082371
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant