pytorch
4306a188 - Fix tied params with Fake Tensor (#85065)

Commit
2 years ago
Fix tied params with Fake Tensor (#85065) Tested locally to fix `USE_FAKE_TENSOR=1 python benchmarks/huggingface.py --ci -d cuda --float32 --backend=aot_nop --training --only=RobertaForCausalLM`. When I tried to repro with a small example test could not successfully, but @anijain2305 is attempting to with minifier. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85065 Approved by: https://github.com/anijain2305
Author
Committer
Parents
Loading