superbench/benchmarks/micro_benchmarks/dist_inference.py · b97ddcf7238accdc069aa0c4b6030f15754b18b2 · tsoc / superbenchmark · GitLab

Find file Blame History Permalink

Fix wrong torch usage in communication wrapper for Distributed Inference Benchmark (#505) · b97ddcf7
Ziyue Yang authored Apr 06, 2023
```
**Description**
This commit fixes wrong `torch.empty_like` usage and missing dtype and
device argument in communication wrappers.
```
b97ddcf7

dist_inference.py 15.9 KB