Add TBE support to benchmark. (#919) · kabicm/FBGEMM@5ae23f9 · GitHub

Commit

Add TBE support to benchmark. (pytorch#919)

Browse files

Summary:
Pull Request resolved: pytorch#919

Add two modes, "P2P_TBE", "P2P_FUSED_TBE", which adds TBE before P2P.

Reviewed By: jianyuh

Differential Revision: D32066265

fbshipit-source-id: 4e1ebc1ff0ff0276a2098af1e77505dcab2a9dba

Loading branch information

caogao authored and facebook-github-bot committed Feb 15, 2022

1 parent f0719cc commit 5ae23f9