mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
Fixes #81018, based on #81036. It will create graph break for cpu 0d tensor value due to .item() call (we could maybe specialize on that instead of breaking?), but otherwise it would create graph break due to synchronizing `to` call, so there's no way around :-(, and for number `value` argument we already should be specializing. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82737 Approved by: https://github.com/Chillee |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| decompositions.py | ||