Skip to content

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

Closed
zoranjovanovic-ns wants to merge 1 commit intoopenxla:mainfrom
ROCm:rocm_triton_backend
Closed

[ROCm] Triton in XLA for ROCm - gemm rewriter triton related changes.#10510
zoranjovanovic-ns wants to merge 1 commit intoopenxla:mainfrom
ROCm:rocm_triton_backend

Conversation

@zoranjovanovic-ns
Copy link
Contributor

First commit of the series for enabling Triton in XLA for ROCm .

@github-actions github-actions bot added the kokoro:force-run Forces CI to rerun label Mar 13, 2024
@kokoro-team kokoro-team removed the kokoro:force-run Forces CI to rerun label Mar 13, 2024
@ddunl ddunl requested a review from tdanyluk March 13, 2024 20:32
Copy link
Contributor

@tdanyluk tdanyluk left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This looks good to me, thank you.

copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 14, 2024
…ed changes.

Imported from GitHub PR openxla/xla#10510

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

--
f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d 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=openxla/xla#10510 from ROCm:rocm_triton_backend f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 14, 2024
…ed changes.

Imported from GitHub PR openxla/xla#10510

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

--
f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d 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=openxla/xla#10510 from ROCm:rocm_triton_backend f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 14, 2024
…ed changes.

Imported from GitHub PR openxla/xla#10510

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

--
f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d 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=openxla/xla#10510 from ROCm:rocm_triton_backend f680e98aa6908d02021bc0c8bb3d24edfc6b2a5d
PiperOrigin-RevId: 615707894
@i-chaochen
Copy link
Contributor

@tdanyluk we notice there is a conflict, but there is a PR from copybara-service to merge now. We wonder shall we resolve this conflict and rebase it again? which I think might need you to approve again?

@cheshire
Copy link
Contributor

Yes, could you re-base and I'll re-approve?

@github-actions github-actions bot added the kokoro:force-run Forces CI to rerun label Mar 15, 2024
@kokoro-team kokoro-team removed the kokoro:force-run Forces CI to rerun label Mar 15, 2024
@zoranjovanovic-ns
Copy link
Contributor Author

Yes, could you re-base and I'll re-approve?

@cheshire rebased and pushed.

copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 15, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 15, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 15, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 15, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 615707894
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 15, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 615707894
@zoranjovanovic-ns
Copy link
Contributor Author

@cheshire Do you know what is blocking PR from being merged?

copybara-service bot pushed a commit that referenced this pull request Mar 18, 2024
…ed changes.

Imported from GitHub PR #10510

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

--
832d725 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=#10510 from ROCm:rocm_triton_backend 832d725
PiperOrigin-RevId: 616862111
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 18, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 616862111
@tdanyluk
Copy link
Contributor

There are some internal failures, but nothing major, I'm trying to submit it.

copybara-service bot pushed a commit that referenced this pull request Mar 20, 2024
…ed changes.

Imported from GitHub PR #10510

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

--
832d725 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=#10510 from ROCm:rocm_triton_backend 832d725
PiperOrigin-RevId: 616862111
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 20, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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=openxla/xla#10510 from ROCm:rocm_triton_backend 832d7253db1f8972862252f034b216d1eed1da29
PiperOrigin-RevId: 616862111
copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Mar 20, 2024
…ed changes.

Imported from GitHub PR openxla/xla#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

PiperOrigin-RevId: 617500238
steeve pushed a commit to zml/xla that referenced this pull request Aug 30, 2024
…n related changes.

Imported from GitHub PR openxla#10510

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

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

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

Merging this change closes openxla#10510

COPYBARA_INTEGRATE_REVIEW=openxla#10510 from ROCm:rocm_triton_backend 832d725
PiperOrigin-RevId: 617500238
@alekstheod alekstheod deleted the rocm_triton_backend branch July 3, 2025 08:11
@alekstheod alekstheod restored the rocm_triton_backend branch July 15, 2025 08:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants