-
Notifications
You must be signed in to change notification settings - Fork 26.3k
make range functions respect current stream #21619
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
Conversation
|
@iotamudelta any advice on fixing ? but I don't need allocator for the ranges, and thrust::cuda::par.on(stream) is valid syntax. |
|
@ngimel looks to me like a shortcoming of the hipified Thrust. Could this be a construct that didn't exist in Thrust some two years ago? Either way, we are working on updating things in that realm, so could you work around it via |
|
rocm builds were fixed by including header, caffe2_onnx failure seems unrelated. BTW, what happened to cuda 10 ci? |
It's run on master, but not on every commit. If you like, I can make it possible to explicitly request cuda 10 CI on pull requests, if you're futzing around with CUDA 10 specific code. |
ezyang
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Very nice.
|
I'm going to go ahead and merge this, but ideally we'd also turn back on the failing tests when we make fixes like this. (This might be hard/annoying to do.) |
facebook-github-bot
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@ezyang is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
But it does not fix the tests :-( Most of the tests fail because of backward running on the wrong stream. This is something that I noticed looking on the profile, and that has to be fixed anyway. |
Summary: Stream is not respected on range/linspace/logspace functions, which contributes to pytorch/pytorch#21589 (this is not a complete solution for that issue). Pull Request resolved: pytorch/pytorch#21619 Differential Revision: D15769666 Pulled By: ezyang fbshipit-source-id: 7c036f7aecb3119430c4d432775cad98a5028fa8
Stream is not respected on range/linspace/logspace functions, which contributes to #21589 (this is not a complete solution for that issue).