-
Notifications
You must be signed in to change notification settings - Fork 685
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
refine_error_info_with_diff_device_input_and_fix_typo #8054
refine_error_info_with_diff_device_input_and_fix_typo #8054
Conversation
CHECK_OR_RETURN(device_tag == JUST(GetDeviceTagOfTensor(input_tensor))) | ||
<< Error::RuntimeError() << "Lazy nn.Graph name: " << graph_name | ||
<< " encountered ERROR in module/op_name: " << new_op_name | ||
<< ". Expected all tensors to be on the same device, but found at least two devices, " | ||
<< JUST(inputs.at(0)->device())->ToString() << " and " | ||
<< JUST(inputs.at(i)->device())->ToString() | ||
<< "! Please use tensor.to() to synchronize all the input with the same device."; |
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.
这里是要保留graph信息还是完全和eager对齐?需要 @strint 确认一下
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.
这里还提供了graph op信息、提供了修改建议。
这样部分对齐就挺好。
https://github.com/Oneflow-Inc/oneflow into refine_error_info_with_diff_device_input_and_fix_typo
…refine_error_info_with_diff_device_input_and_fix_typo
https://github.com/Oneflow-Inc/oneflow into refine_error_info_with_diff_device_input_and_fix_typo
Speed stats:
|
CI failed when running job: Build cu102_xla. PR label automerge has been removed |
View latest API docs preview at: https://staging.oneflow.info/docs/Oneflow-Inc/oneflow/pr/8054/ |
Speed stats:
|
当遇到 多个 input placement 或则不一致时,报错信息对齐pytorch