xla icon indicating copy to clipboard operation
xla copied to clipboard

[XLA:GPU] Pass the CUDA / ROCm toolkit version explicitly for autotuning and GEMM rewriting.

Open copybara-service[bot] opened this issue 1 year ago • 0 comments

[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.

copybara-service[bot] avatar May 13 '24 13:05 copybara-service[bot]