github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Resolve Mac torch-mlir torch setup dependency. Enable MacOS CI

Browse Source
This commit is contained in:
PhaneeshB
2022-11-07 13:16:24 +05:30
committed by Prashant Kumar
parent 1e95e4b502
commit 64d6da75c7
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/test-models.yml vendored
View File

@@ -36,8 +36,6 @@ jobs:
suite: cuda suite: cuda
- os: ubuntu-latest - os: ubuntu-latest
suite: cpu suite: cpu
- os: MacStudio
suite: vulkan
- os: MacStudio - os: MacStudio
suite: cuda suite: cuda
- os: MacStudio - os: MacStudio

5
setup_venv.sh
View File

@@ -77,7 +77,8 @@ $PYTHON -m pip install --upgrade pip || die "Could not upgrade pip"
$PYTHON -m pip install --upgrade -r "$TD/requirements.txt" $PYTHON -m pip install --upgrade -r "$TD/requirements.txt"
if [ "$torch_mlir_bin" = true ]; then if [ "$torch_mlir_bin" = true ]; then
if [[ $(uname -s) = 'Darwin' ]]; then if [[ $(uname -s) = 'Darwin' ]]; then
echo "MacOS detected. Please install torch-mlir from source or .whl, as dependency problems may occur otherwise." echo "MacOS detected. Installing torch-mlir from .whl, to avoid dependency problems with torch."
$PYTHON -m pip install --pre --no-cache-dir torch-mlir -f https://llvm.github.io/torch-mlir/package-index/ -f https://download.pytorch.org/whl/nightly/torch/
else else
$PYTHON -m pip install --pre torch-mlir -f https://llvm.github.io/torch-mlir/package-index/ $PYTHON -m pip install --pre torch-mlir -f https://llvm.github.io/torch-mlir/package-index/
if [ $? -eq 0 ];then if [ $? -eq 0 ];then
@@ -104,7 +105,7 @@ else
echo "Not installing a backend, please make sure to add your backend to PYTHONPATH" echo "Not installing a backend, please make sure to add your backend to PYTHONPATH"
fi fi
$PYTHON -m pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f ${RUNTIME} $PYTHON -m pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f ${RUNTIME} -f https://download.pytorch.org/whl/nightly/torch/
if [[ ! -z "${IMPORTER}" ]]; then if [[ ! -z "${IMPORTER}" ]]; then
echo "${Yellow}Installing importer tools.." echo "${Yellow}Installing importer tools.."