diff options
| author | ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-12 11:18:11 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-12 11:18:11 +0100 |
| commit | e90bcdcc7c51dd85037055b59f22568100d801f0 (patch) | |
| tree | 2c7e89df98f44192d92b185682d40b71295fd704 /candle-pyo3/test_pytorch.py | |
| parent | 8e06bfb4fd33f1229a03abee20cc1c07198408b5 (diff) | |
| download | candle-e90bcdcc7c51dd85037055b59f22568100d801f0.tar.gz candle-e90bcdcc7c51dd85037055b59f22568100d801f0.tar.bz2 candle-e90bcdcc7c51dd85037055b59f22568100d801f0.zip | |
Metal: f16 and bf16 where_cond + benchmark (#1545)
* Use cfg to seperate benchmark results based on features
* Add metal where_cond for f16 and bf16. Add benchmark
* Remove allow pragma
* Avoid some unnecessary returns.
* Improve benchmarks layout
* Updated feature separated benchmarks
---------
Co-authored-by: Laurent <laurent.mazare@gmail.com>
Diffstat (limited to 'candle-pyo3/test_pytorch.py')
0 files changed, 0 insertions, 0 deletions