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

[PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.

Browse Source 
The two are aliases and the former is a forwarding header pointing to the latter.

Cleanup only, no functional changes.

PiperOrigin-RevId: 621341188
This commit is contained in:
Peter Hawkins 2024-04-02 17:19:24 -07:00 committed by jax authors
parent 318ae8935a
commit 1baed9b285
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/jax_cpp/BUILD
View File

@ -27,7 +27,7 @@ cc_binary(
"@xla//xla:status",
"@xla//xla:statusor",
"@xla//xla/pjrt:pjrt_client",
"@xla//xla/pjrt:tfrt_cpu_pjrt_client",
"@xla//xla/pjrt/cpu:cpu_client",
"@xla//xla/service:hlo_proto_cc",
"@xla//xla/tools:hlo_module_loader",
"@tsl//tsl/platform:logging",

10
examples/jax_cpp/main.cc
View File

@ -27,10 +27,10 @@ limitations under the License.
//
// To load and run the HloModule,
//
// $ bazel build examples/jax_cpp:main --experimental_repo_remote_exec --check_visibility=false
// $ bazel-bin/examples/jax_cpp/main
// 2021-01-12 15:35:28.316880: I examples/jax_cpp/main.cc:65] result = (
// f32[2,2] {
// $ bazel build examples/jax_cpp:main --experimental_repo_remote_exec \
// --check_visibility=false
// $ bazel-bin/examples/jax_cpp/main 2021-01-12
// 15:35:28.316880: I examples/jax_cpp/main.cc:65] result = ( f32[2,2] {
// { 1.5, 1.5 },
// { 3.5, 3.5 }
// }
@ -42,8 +42,8 @@ limitations under the License.
#include "xla/literal.h"
#include "xla/literal_util.h"
#include "xla/pjrt/cpu/cpu_client.h"
#include "xla/pjrt/pjrt_client.h"
#include "xla/pjrt/tfrt_cpu_pjrt_client.h"
#include "xla/status.h"
#include "xla/statusor.h"
#include "xla/tools/hlo_module_loader.h"