Comments (2)
Currently the XLA Python bindings are something of an internal detail of JAX and they aren't exposed at all as part of TensorFlow. It's not clear whether they will remain part of XLA itself or move into JAX; that is to be determined.
Can you say more about the use case? That might help answer the question about public APIs.
from xla.
Understood.
Here is the usecase :
We would like to handle exceptions arising from XLA with custom messages in the framework. But XlaRuntimeError is not visible in tensorflow python layer. What are the steps needed to make it visible in tensorflow?
from xla.
Related Issues (20)
- [XLA:CPU] oneDNN Softmax gives inaccurate results HOT 1
- call ptxas become defunct, cause xla hung HOT 2
- gpu_hlo_cost_analysis NumOfDevices always return 0 HOT 2
- [Feature Request]Add more comm op support on gpu_hlo_cost_analysis HOT 4
- gpu f16 cast to fp32 calculation, and then converted back? HOT 1
- "Could not find executable `nvidia-smi`" for `./configure.py --backend=CUDA` HOT 6
- Build from source fails HOT 7
- XLA documentation for Windows HOT 1
- Implement GitHub Presubmit Checks for Windows Environment HOT 1
- Support builds with cuDNN 9
- gpu_fused_mha_test fails at HEAD on H100 HOT 1
- Workable example on normal DNN model
- [xla:auto_sharding] Question about resharding costs of Reshape strategies
- OpenCL Support. HOT 2
- BF16 matmul slower than F32 matmul on T4 GPU HOT 3
- PJRT `CopyCpuBufferToLiteral` of JAX buffer taking too long HOT 9
- Porting XLA to different backends. HOT 4
- Compiling xla/mlir/tools/mlir_interpreter/dialects/util.cc failed HOT 3
- MHLO Extraction from XLA Compiler HOT 4
- Controlling a single compiler pass in XLA for CPU target HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from xla.