tests/benchmarks/model_benchmarks/test_pytorch_base.py · f944245694de3309e7f6c8cf773a9a9e493297d8 · tsoc / superbenchmark · GitLab

Find file Blame History Permalink

Benchmarks: Add Feature - Add option to use fp32 instead of tf32 (#213) · f9442456
guoshzhao authored Sep 28, 2021
```
**Description**
Add option `force_fp32` to use fp32 instead of tf32, only takes effect on Ampere or newer GPUs.
```
f9442456

test_pytorch_base.py 9.55 KB