xla icon indicating copy to clipboard operation
xla copied to clipboard

[Draft][XLA:GPU] Add sycl runtime

Open ShengYang1 opened this issue 1 year ago • 2 comments

This PR aims to add sycl runtime support, we can run basic JAX GPU UTs with it. It includes:

  1. sycl runtime crosstool build
  2. sycl stream executor
  3. spirv-llvm-translator
  4. registration of spir_compiler/computation placer/transfer manager
  5. refine build file to remove cuda specific changes(especially those cuda kernels)
  6. fix CudaCubin to load spirv binary
  7. fix metadata

This is a draft PR to show what is needed to run JAX on sycl runtime.

ShengYang1 avatar Feb 01 '24 06:02 ShengYang1