mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ROCM] Replace xla::Status with absl::Status in rocm_plugin_extension.cc

Browse Source
This commit is contained in:
Ruturaj4 2024-07-02 10:45:48 -05:00
parent 42724ebc73
commit 4b936233a6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/rocm_plugin_extension.cc
View File

@ -34,7 +34,7 @@ namespace nb = nanobind;
namespace xla {
namespace {
Status RegisterCustomCallTarget(const PJRT_Api* c_api, nb::str fn_name,
absl::Status RegisterCustomCallTarget(const PJRT_Api* c_api, nb::str fn_name,
nb::capsule fn, int api_version,
XLA_FFI_Handler_Traits traits) {
if (c_api->extension_start == nullptr) {
@ -66,7 +66,7 @@ Status RegisterCustomCallTarget(const PJRT_Api* c_api, nb::str fn_name,
RETURN_STATUS_IF_PJRT_ERROR(
reinterpret_cast<const PJRT_Gpu_Custom_Call*>(next)->custom_call(&args),
c_api);
return OkStatus();
return absl::OkStatus();
}
nb::dict Registrations() {