PR #10510: [ROCm] Triton in XLA for ROCm - gemm rewriter triton related changes.
PR #10510: [ROCm] Triton in XLA for ROCm - gemm rewriter triton related changes.

Imported from GitHub PR https://github.com/openxla/xla/pull/10510

First commit of the series for enabling Triton in XLA for ROCm .
Copybara import of the project:

--
832d7253db1f8972862252f034b216d1eed1da29 by Zoran Jovanovic <zjovanov@amd.com>:

[ROCm] Triton in XLA for ROCm - gemm rewriter triton related changes.

Merging this change closes #10510

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
