pytorch
39201ce0 - Make dynamo bench conditionally import DDP/FSDP (#103163)

Comment changes are shownComment changes are hidden
Commit
1 year ago
Make dynamo bench conditionally import DDP/FSDP (#103163) Avoids hitting importerror for singlenode benchmarks when running on a non-distributed build of pytorch. Fixes #102086 Pull Request resolved: https://github.com/pytorch/pytorch/pull/103163 Approved by: https://github.com/lezcano, https://github.com/wanchaol
Author
Committer
Parents
  • benchmarks/dynamo
    • File
      common.py