[AutoParallel] fix intermediate api pipe hook tuple object bug#75081
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
Codecov Report✅ All modified and coverable lines are covered by tests. Additional details and impacted files@@ Coverage Diff @@
## develop #75081 +/- ##
===========================================
Coverage ? 100.00%
===========================================
Files ? 1
Lines ? 7
Branches ? 0
===========================================
Hits ? 7
Misses ? 0
Partials ? 0 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
|
/re-run all-failed |
| else: | ||
| raise ValueError( | ||
| f"output should be a dict of tensors or list of tensors or tensor, but {type(output)}" | ||
| f"output should be a dict/tensors/list of tensors/tuple of tensor/tensor, but {type(output)}" |
There was a problem hiding this comment.
这行报错，是什么意思 output should be a dict/tensors/list of tensors/tuple of tensor/tensor
There was a problem hiding this comment.
pp stage之间传输的值只应该是这几个类型之一
|
/re-run all-failed |
3 similar comments
|
/re-run all-failed |
|
/re-run all-failed |
|
/re-run all-failed |
|
/re-run all-failed |
2 similar comments
|
/re-run all-failed |
|
/re-run all-failed |
PR Category
Auto Parallel
PR Types
Bug fixes
Description
在使用中层api进行pipeline时，会使用pipe_hook插入在不同的pipe stage之间传输数据的层，当该数据（output）为tuple时，会触发 'tuple' object does not support item assignment；该pr针对tuple output单独处理来解决该bug