/stardew

Primary LanguagePython

Stardew

Bunch of experiments on lowering PyTorch/Tensorflow to MLIR.

Dev Setup

Create Virtual Environment

This was tested on python 3.11.3. The project requires Python >= 3.11.

python -m venv .env
source .env/bin/activate
pip install -r requirements.txt --require-venv

Now for PyTorch experiments:

pip install -r requirements-torch.txt --require-venv

And/Or Tensorflow experiments:

pip install -r requirements-tf.txt --require-venv

Install Stardew

pip install --editable . --require-venv

Running an example

To run the examples, you need pytest:

pip install pytest

Then you can run the tests with:

pytest tests