Skip to content

[inductor] fix: 'get_raw_stream' undefined#163707

Closed
dolpm wants to merge 1 commit intopytorch:mainfrom
dolpm:remote/inductor-get-raw-stream-undefined
Closed

[inductor] fix: 'get_raw_stream' undefined#163707
dolpm wants to merge 1 commit intopytorch:mainfrom
dolpm:remote/inductor-get-raw-stream-undefined

Conversation

@dolpm
Copy link
Contributor

@dolpm dolpm commented Sep 23, 2025

Summary:
ran into this when precompiling baidu/ERNIE-4.5-21B-A3B-PT

codegen after fix:

import triton
import triton.language as tl
from torch._inductor.runtime.triton_heuristics import start_graph, end_graph
from torch._C import _cuda_getCurrentRawStream as get_raw_stream
with torch.cuda._DeviceGuard(0):
    stream0 = get_raw_stream(0)
...

cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @chenyang78 @kadeng @muchulee8 @amjames @chauhang @aakhundov @coconutruben

@pytorch-bot
Copy link

pytorch-bot bot commented Sep 23, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/163707

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit bbd312d with merge base 4c2c401 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

…ream' is not defined

Summary:
ran into this when precompiling baidu/ERNIE-4.5-21B-A3B-PT
@dolpm dolpm force-pushed the remote/inductor-get-raw-stream-undefined branch from 8de6216 to bbd312d Compare September 24, 2025 16:58
@jamesjwu
Copy link
Contributor

cc @eellison for context but LGTM

@dolpm
Copy link
Contributor Author

dolpm commented Sep 25, 2025

@pytorchmergebot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Sep 25, 2025
@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@pytorchmergebot
Copy link
Collaborator

Merge failed

Reason: 1 jobs have failed, first few of them are: trunk / inductor-build / build

Details for Dev Infra team Raised by workflow job

@dolpm
Copy link
Contributor Author

dolpm commented Sep 29, 2025

@pytorchmergebot merge

@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants