mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
773e6ebebf
* Move IREE pins for windows, linux, and pip packaging files. * Remove linux builds from 1.0 nightly workflow It seems that the VMs used for these workflows are no longer available. Removing linux builds since publishing .exe is sufficient for the one-shot nightly workflows we trigger for SHARK-1.0
98 lines
2.8 KiB
PowerShell
98 lines
2.8 KiB
PowerShell
<#
|
|
.SYNOPSIS
|
|
A script to update and install the SHARK runtime and its dependencies.
|
|
|
|
.DESCRIPTION
|
|
This script updates and installs the SHARK runtime and its dependencies.
|
|
It checks the Python version installed and installs any required build
|
|
dependencies into a Python virtual environment.
|
|
If that environment does not exist, it creates it.
|
|
|
|
.PARAMETER update-src
|
|
git pulls latest version
|
|
|
|
.PARAMETER force
|
|
removes and recreates venv to force update of all dependencies
|
|
|
|
.EXAMPLE
|
|
.\setup_venv.ps1 --force
|
|
|
|
.EXAMPLE
|
|
.\setup_venv.ps1 --update-src
|
|
|
|
.INPUTS
|
|
None
|
|
|
|
.OUTPUTS
|
|
None
|
|
|
|
#>
|
|
|
|
param([string]$arguments)
|
|
|
|
if ($arguments -eq "--update-src"){
|
|
git pull
|
|
}
|
|
|
|
if ($arguments -eq "--force"){
|
|
if (Test-Path env:VIRTUAL_ENV) {
|
|
Write-Host "deactivating..."
|
|
Deactivate
|
|
}
|
|
|
|
if (Test-Path .\shark1.venv\) {
|
|
Write-Host "removing and recreating venv..."
|
|
Remove-Item .\shark1.venv -Force -Recurse
|
|
if (Test-Path .\shark1.venv\) {
|
|
Write-Host 'could not remove .\shark-venv - please try running ".\setup_venv.ps1 --force" again!'
|
|
exit 1
|
|
}
|
|
}
|
|
}
|
|
|
|
# redirect stderr into stdout
|
|
$p = &{python -V} 2>&1
|
|
# check if an ErrorRecord was returned
|
|
$version = if($p -is [System.Management.Automation.ErrorRecord])
|
|
{
|
|
# grab the version string from the error message
|
|
$p.Exception.Message
|
|
}
|
|
else
|
|
{
|
|
# otherwise return complete Python list
|
|
$ErrorActionPreference = 'SilentlyContinue'
|
|
$PyVer = py --list
|
|
}
|
|
|
|
# deactivate any activated venvs
|
|
if ($PyVer -like "*venv*")
|
|
{
|
|
deactivate # make sure we don't update the wrong venv
|
|
$PyVer = py --list # update list
|
|
}
|
|
|
|
Write-Host "Python versions found are"
|
|
Write-Host ($PyVer | Out-String) # formatted output with line breaks
|
|
if (!($PyVer.length -ne 0)) {$p} # return Python --version String if py.exe is unavailable
|
|
if (!($PyVer -like "*3.11*") -and !($p -like "*3.11*")) # if 3.11 is not in any list
|
|
{
|
|
Write-Host "Please install Python 3.11 and try again"
|
|
exit 34
|
|
}
|
|
|
|
Write-Host "Installing Build Dependencies"
|
|
# make sure we really use 3.11 from list, even if it's not the default.
|
|
if ($NULL -ne $PyVer) {py -3.11 -m venv .\shark1.venv\}
|
|
else {python -m venv .\shark1.venv\}
|
|
.\shark1.venv\Scripts\activate
|
|
python -m pip install --upgrade pip
|
|
pip install wheel
|
|
pip install -r requirements.txt
|
|
pip install --pre torch-mlir torchvision torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://llvm.github.io/torch-mlir/package-index/
|
|
pip install --upgrade -f https://nod-ai.github.io/SRT/pip-release-links.html iree-compiler==20231212.* iree-runtime==20231212.*
|
|
Write-Host "Building SHARK..."
|
|
pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://nod-ai.github.io/SRT/pip-release-links.html
|
|
Write-Host "Build and installation completed successfully"
|
|
Write-Host "Source your venv with ./shark1.venv/Scripts/activate"
|