Skip to content

[CuteDSL] Fix source location info for cute.arch.elect_one#3317

Open
pchen7e2 wants to merge 1 commit into
NVIDIA:mainfrom
pchen7e2:fix-elect-op-loc
Open

[CuteDSL] Fix source location info for cute.arch.elect_one#3317
pchen7e2 wants to merge 1 commit into
NVIDIA:mainfrom
pchen7e2:fix-elect-op-loc

Conversation

@pchen7e2

Copy link
Copy Markdown

The loc info was missed when creating nvvm.elect_sync. This means the later pipelines would not be able to tell that NVVM Op was from "this line of kernel source". The fix is to attach it.

We can check the effect by looking at source mapping in ncu:

Before: (elect.sync was not mapped correctly. In fact it's mapped to a line @cute.jit)

Screenshot 2026-06-14 at 15 33 51

After: (elect.sync was correctly mapped)

Screenshot 2026-06-14 at 15 35 36

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