mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
Setup packaging
This commit is contained in:
12
pyproject.toml
Normal file
12
pyproject.toml
Normal file
@@ -0,0 +1,12 @@
|
||||
[build-system]
|
||||
requires = [
|
||||
"setuptools>=42",
|
||||
"wheel",
|
||||
"packaging",
|
||||
|
||||
"numpy",
|
||||
"torch-mlir",
|
||||
"iree-compiler",
|
||||
"iree-runtime",
|
||||
]
|
||||
build-backend = "setuptools.build_meta"
|
||||
8
setup.py
8
setup.py
@@ -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",
|
||||
|
||||
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user