Add PTX CompilationProvider for compiling PTX via the CUDA driver

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.74k stars 440 forks source link

Closed copybara-service[bot] closed 3 days ago

copybara-service[bot] commented 4 days ago

Add PTX CompilationProvider for compiling PTX via the CUDA driver

Adds a new compilation provider for driver compilation.
Moves compilation log output parser logic into ptx_compiler_helpers and use the shared logic in both driver and nvjitlink compilation.
Adds test for the log output parser logic.
Splits the compilation_provider_test into 2 test targets - one which requires a GPU and one which doesn't