superbench/benchmarks/model_benchmarks/pytorch_base.py · c770ed5de27018f71fb2f4e288343a3905d63511 · tsoc / superbenchmark

Benchmarks: Revise Code - Add support for pytorch>=1.9.0 of init_process_group (#305) · e31b8c9e

user4543 authored Feb 10, 2022

**Description**
Add support for pytorch>=1.9.0 of init_process_group.

**Major Revision**
- Use PrefixStore(TCPStore) to init_process_group manully for each model run

e31b8c9e

pytorch_base.py 9.84 KB

Replace pytorch_base.py