xla
xla copied to clipboard
[Draft][XLA:GPU] Add sycl runtime
This PR aims to add sycl runtime support, we can run basic JAX GPU UTs with it. It includes:
- sycl runtime crosstool build
- sycl stream executor
- spirv-llvm-translator
- registration of spir_compiler/computation placer/transfer manager
- refine build file to remove cuda specific changes(especially those cuda kernels)
- fix CudaCubin to load spirv binary
- fix metadata
This is a draft PR to show what is needed to run JAX on sycl runtime.