Setup packaging

This commit is contained in:
anush elangovan
2022-04-26 00:22:02 -07:00
parent cb67b0b3f1
commit 35b1e33108
3 changed files with 19 additions and 3 deletions

12
pyproject.toml Normal file
View File

@@ -0,0 +1,12 @@
[build-system]
requires = [
"setuptools>=42",
"wheel",
"packaging",
"numpy",
"torch-mlir",
"iree-compiler",
"iree-runtime",
]
build-backend = "setuptools.build_meta"

View File

@@ -1,11 +1,15 @@
from setuptools import find_packages
from setuptools import setup
import os
PACKAGE_VERSION = os.environ.get("SHARK_PACKAGE_VERSION") or "0.0.1"
setup(
name="shark",
version="0.0.1",
description="The Shark Runner provides inference and training APIs to run deep learning models on Shark Runtime.",
version=f"{PACKAGE_VERSION}",
description="The SHARK Runner provides inference and training APIs to run deep learning models on Shark Runtime.",
author="Nod Labs",
author_email="stdin@nod.com",
url="https://github.com/NodLabs/dSHARK",

View File

@@ -34,5 +34,5 @@ python -m pip install .
Red=`tput setaf 1`
Green=`tput setaf 2`
echo "${Green}Before running examples activate venv with:"
echo " ${Red}source $VENV_DIR/bin/activate"
echo " ${Green}source $VENV_DIR/bin/activate"