is tensorflow xla/python maintained still?

112 views
Skip to first unread message

Mingting Sun

unread,
May 17, 2021, 6:55:16 AM5/17/21
to XLA development
Hello XLA team

I want to run xla/python as test cases, but I find lots of compile error.
I wonder know, is xla/python code still maintained?

Or I need turn to use JAX for XLA python API?

any suggestions is appreciated, 
thanks in advance.


Best regards !
Mingting Sun

Peter Hawkins

unread,
May 18, 2021, 5:43:17 PM5/18/21
to Mingting Sun, XLA development
Hi...

The XLA:Python APIs are maintained, but they are semi-private APIs maintained primarily for the use of JAX.

One tip is that you can actually access these APIs via jax, although note we don't promise any form of support for them. If you have a working JAX installation, then jax.lib.xla_client is a public entry point for them.

Peter

--
You received this message because you are subscribed to the Google Groups "XLA development" group.
To unsubscribe from this group and stop receiving emails from it, send an email to xla-dev+u...@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/xla-dev/CAD8YVK%2Bjs6D-%3DidS-KWR9_W8nyUB%2BN6XL8XPPd5xNySKiU1o9g%40mail.gmail.com.

Mingting Sun

unread,
May 18, 2021, 9:43:00 PM5/18/21
to Peter Hawkins, XLA development
Hi Peter

Thanks for your information.
I will try jax.

BTW,  sorry for I didn't make "xla/python" clearly.
I meant tensorflow source code, "tensorflow/tensorflow/compiler/xla/python" when I say "xla/python"
I find I cannot compile xla/python  successfully via bazel build or bazel test with TensorFlow 1.15 release.

fortunately, I find there is API in TensorFlow, after import xla package under tf2xla, it works and meet my requirements.
"from tensorflow.compiler.tf2xla.python import xla"
 


Best regards !
Mingting Sun


Peter Hawkins <phaw...@google.com> 于2021年5月19日周三 上午5:43写道:
Reply all
Reply to author
Forward
0 new messages