fix: fix the "schema not found for node" error
Signed-off-by: Bo Wang [email protected]
Description
In partitioning, when we check if a node would modify the input value, we use the node->schema() function from PyTorch previously. However, this function could return a nullptr in some cases, which triggers a error showing Schema not found for node. File a bug report
Fixes #1227
Type of change
- Bug fix (non-breaking change which fixes an issue)
Checklist:
- [ ] My code follows the style guidelines of this project (You can use the linters)
- [ ] I have performed a self-review of my own code
- [ ] I have commented my code, particularly in hard-to-understand areas and hacks
- [ ] I have made corresponding changes to the documentation
- [ ] I have added tests to verify my fix or my feature
- [ ] New and existing unit tests pass locally with my changes
- [ ] I have added the relevant labels to my PR in so that relevant reviewers are notified
@frank-wei @narendasan It looks like FX tests test-py-fx-x86_64-pyt-release are failing with the following error. Is this a known issue ?
RuntimeError: CUDA error: an illegal memory access was encountered
@frank-wei @narendasan It looks like FX tests test-py-fx-x86_64-pyt-release are failing with the following error. Is this a known issue ?
RuntimeError: CUDA error: an illegal memory access was encountered
I merged my fix for nightly yesterday. Did you rebase on master?
How can i intstall the latest version in my docker ? the latest released docker (22.08) seems haven't fixed the bug yet. @narendasan