xla
xla copied to clipboard
[XLA:GPU] Pass the CUDA / ROCm toolkit version explicitly for autotuning and GEMM rewriting.
[XLA:GPU] Pass the CUDA / ROCm toolkit version explicitly for autotuning and GEMM rewriting.
This allows to remove more #if GOOGLE_CUDA preprocessor directives from HLO passes.