pytorch-benchmark - Update CUDA benchmarking with best Events and syncronize

What's Changed

• Cuda sync by @LukasHedegaard in https://github.com/LukasHedegaard/pytorch-benchmark/pull/15
• Updated on-gpu benchmarking by @LukasHedegaard in https://github.com/LukasHedegaard/pytorch-benchmark/pull/16

Full Changelog: https://github.com/LukasHedegaard/pytorch-benchmark/compare/0.3.5...0.3.6

- Python
Published by LukasHedegaard almost 3 years ago

pytorch-benchmark - Add DOI

This release adds a DOI to the codebase

- Python
Published by LukasHedegaard over 3 years ago

pytorch-benchmark - Add missing memory to results

Fixed

• Add missing memory to results.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Fix memory measurement for batch size 1

Fixed

• Memory measurement for bs=1

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Fix batch size during warmup

Fixed

• Warm up batch size

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Support for FLOPs in nn.Modules with input other than tensor

 Removed

• try_custom_warmup.

Added

• warm_up_fn overload option.
• Support for FLOPs count in torch.nn.Module with input other than Tensor.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Per batch_size memory measurement and repeated energy measurements

Added

• Memory measurement for each batch size.
• Repeated energy measurement.

Changed

• Number formatting to use u instead of µ.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Print options

Added

• Option to redirect info prints.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Fix gradient tracking

Fixed

• Added missing with torch.no_grad()

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Support for custom classes

Added

• Overloads for benchmark parameters and functions to allow benchmark of custom classes.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - GPU fixes

Fixed

• GPU compatibility.

Removed

• Carbon-tracker energy measurement. Library is still too immature at this point.

- Python
Published by LukasHedegaard over 4 years ago

pytorch-benchmark - Initial release

Added

• Initial version.

- Python
Published by LukasHedegaard over 4 years ago