История

Karla Saur 3edbfb7566 flake8-6.1.0 adds E721 check (#730 )		2023-08-07 12:43:13 -07:00
..
README.md	Add scripts for the operators bench (#330 )	2020-10-23 15:40:46 -07:00
__init__.py	Add scripts for the operators bench (#330 )	2020-10-23 15:40:46 -07:00
run.py	Add TVM backend (#236 )	2020-11-03 13:21:02 -08:00
score.py	flake8-6.1.0 adds E721 check (#730 )	2023-08-07 12:43:13 -07:00
train.py	Adding hash to save/load (#713 )	2023-06-20 17:43:26 -07:00

README.md

Operators Experiments

This directory contains the script to reproduce the experiments of Section 6.1.2 of the paper A Tensor Compiler for Unified Machine Learning Prediction Serving. This script is configured to run sklearn and compare it against onnx-ml, torchscript and onnx (the last 2 using Hummingbird), for the iris dataset over 1 core, and with batch of 1M.

python run.py will run the benchmarks for CPU
python run.py -gpu will run the benchmarks for GPU