openxla / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

Home Page:http://iree.dev/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

add iree-benchmark-module to the python package (and other features)

powderluv opened this issue · comments

Add iree-benchmark-module into the python package just like iree-benchmark-trace. This allows for various customers to compare Python overhead between their Python invocation and underlying benchmark-module invocation.

Today the package includes:

./lib/python3.7/site-packages/iree/runtime/iree-run-module
./lib/python3.7/site-packages/iree/runtime/iree-benchmark-trace
./lib/python3.7/site-packages/iree/runtime/iree-run-trace
./lib/python3.7/site-packages/iree/runtime/scripts/iree_run_module
./lib/python3.7/site-packages/iree/runtime/scripts/iree_benchmark_trace
./lib/python3.7/site-packages/iree/runtime/scripts/iree_run_trace
./lib/python3.7/site-packages/iree/runtime/scripts/iree_tracy_capture```

while we are there can we please update the wheel package version to be gitSHA+dirty if not set to something. Right now it defaults to 0.1dev1

(base) anush@MacBook-Pro iree % pip wheel iree/compiler/
Processing ./iree/compiler
  Installing build dependencies ... done
  Getting requirements to build wheel ... done
  Preparing metadata (pyproject.toml) ... done
Collecting PyYAML
  File was already downloaded /Users/anush/github/iree/PyYAML-6.0-cp39-cp39-macosx_11_0_arm64.whl
Collecting numpy
  File was already downloaded /Users/anush/github/iree/numpy-1.22.3-cp39-cp39-macosx_11_0_arm64.whl
Building wheels for collected packages: iree-compiler
  Building wheel for iree-compiler (pyproject.toml) ... done
  Created wheel for iree-compiler: filename=iree_compiler-0.1.dev1-cp39-cp39-macosx_12_0_arm64.whl size=33192681 sha256=c9b6fbbdbf92f0d4dbb736fc0833a05da5ef6fbe38028f77fb4d5e360927f422
  Stored in directory: /private/var/folders/y5/6qfcmgcn2c1c239n7hbgqmx40000gn/T/pip-ephem-wheel-cache-kl1m1dyw/wheels/a5/4c/05/37ec158bd9ade6c7e6ced1322142cbe6b35b45748a2d3ff8f7
Successfully built iree-compiler
(base) anush@MacBook-Pro iree % find . -name '*.whl'                                               
./iree_runtime-0.1.dev1-cp39-cp39-macosx_12_0_arm64.whl
./compiler-0.2.0-py3-none-any.whl
./iree_compiler-0.1.dev1-cp39-cp39-macosx_12_0_arm64.whl