search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.55k stars 394 forks source link

PR #14605: [ROCm] Switch on Triton feature for ROCm. #14906

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

PR #14605: [ROCm] Switch on Triton feature for ROCm.

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

Last in series of commits to switch on Triton in XLA for ROCm.

This is new version of: https://github.com/openxla/xla/pull/13003

Changes in third_party/triton/temporary/amd_pr7.patch are already merged on: https://github.com/triton-lang/triton/pull/4238 Copybara import of the project:

-- c2ce7e08180225a8e1433b013ded16ebabf25615 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Switch on Triton feature for ROCm.

-- 563b3036893fccbd72f11ae6a80e110c516bd0ac by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Fixed an issue with test cases from ir_emitter_triton_test.cc

-- a4d2ad85f251b37f75d8b4f80aeba075932e6777 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Fixed an issue with gpu_compiler_test.cc

-- a1b9260a0e54e74ab5dd363e8b755f3534634064 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Applied comments from code review.

-- c694a95420b2bb73871389b5d3dd53c97e3e50da by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Fixed failed tests because of https://github.com/openxla/xla/commit/19c11baa83f31e25a3f841cf41fa47a53e8ca161

-- 7359619ba75d388817c07f30c0ee1094a1c11184 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Fixed compilation issue with latest rebase.

-- 82f58ce245c44065df2983c3f6585c3b8c536914 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Skip SplitLHSInputOutputIsFused test in ir_emitter_triton_test.cc untill issue is fixed.

-- 57e776be2336e5a99f587390e35705dbf59277c8 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Triton related changes merged thus removed amd_pr7.patch

-- 0d09d0e283d0210cac7f368d550a5e0e7e2d3a8b by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Applied comments from code review.

-- 7b111478768dc267ec1eed8c3555ed1d1070791f by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Applied comments from code review.

-- 9e7e0c7c05185fc476383b654744e4b2e5079da1 by Zoran Jovanovic zjovanov@amd.com:

[ROCm] Modified TestNoAutotuner test case.

Merging this change closes #14605

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/14605 from ROCm:rocm_triton_backend_8 9e7e0c7c05185fc476383b654744e4b2e5079da1