Skip to content

Fix Device Segfault #152

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

Merged
merged 4 commits into from
Jul 14, 2025
Merged

Fix Device Segfault #152

merged 4 commits into from
Jul 14, 2025

Conversation

micmelesse
Copy link
Collaborator

If tensors are on a specific device such device 7 and triton launches a kernel on the default device such as device 0. We get a segfault. I add asserts to catch this bug. I also add a bunch of other asserts to insure we are working well formed inputs.

@micmelesse
Copy link
Collaborator Author

Main suite passes

image

@micmelesse
Copy link
Collaborator Author

Internal tests pass.

image

@micmelesse
Copy link
Collaborator Author

No change in perf
image

@micmelesse micmelesse marked this pull request as ready for review July 14, 2025 18:50
@micmelesse micmelesse merged commit 5db9d18 into main_perf Jul 14, 2025
1 check failed
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.

1 participant