Automatically profile and select the best meta parameters for NKI kernels.
- Follow AWS Neuron tutorial to install Neuron packages.
pip install -r requirements.txt
- Install NKIPy.
- Install autotune.
pre-commit install
pip install -e .
Make sure you have the necessary build tools:
pip install --upgrade pip wheel build
Build the wheel:
python -m build --wheel
This will create a .whl file in the dist/ directory, something like dist/nki_autotune-0.1.0a0-py3-none-any.whl.
See CONTRIBUTING for more information.
This project is licensed under the Apache-2.0 License.