Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[backend][amd] Add support for device assert #3552

Merged
merged 4 commits into from
Apr 3, 2024

Conversation

antiagainst
Copy link
Collaborator

This commit adds support for device assert in the AMD
backend. This creates a new virtual method on TargetInfo
so that we can specialize for different backends, given
that on AMD side we don't have the same lowering flow:
right now __assert_fail is not supported in the AMDGPU
backend; it's in the runtime.

ROCm/clr@7137a29

Instead we need to layer it on top of print and do trap
afterwards.

@zahimoud
Copy link
Contributor

zahimoud commented Apr 3, 2024

LGTM!

@zahimoud zahimoud merged commit 439d5a1 into triton-lang:main Apr 3, 2024
5 checks passed
@antiagainst antiagainst deleted the amd-device-print branch April 3, 2024 18:32
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.

2 participants