Skip to content

Commit

Permalink
Remove torchcompat override
Browse files Browse the repository at this point in the history
  • Loading branch information
pierre.delaunay committed Jun 11, 2024
1 parent 69baffe commit 5a13ead
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 12 deletions.
3 changes: 0 additions & 3 deletions scripts/article/run_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@ install_prepare() {
if [ ! -d "$MILABENCH_WORDIR/milabench" ]; then
git clone https://github.com/mila-iqia/milabench.git -b intel
git clone https://github.com/Delaunay/voir.git -b async_timer
git clone https://github.com/Delaunay/torchcompat.git
fi

. $MILABENCH_WORDIR/env/bin/activate
Expand All @@ -33,13 +32,11 @@ install_prepare() {

which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat

(
. $BENCHMARK_VENV/bin/activate
which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat
pip install torch torchvision torchaudio

# DALI stuff
Expand Down
3 changes: 0 additions & 3 deletions scripts/article/run_hpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@ install_prepare() {

git clone https://github.com/mila-iqia/milabench.git -b intel
git clone https://github.com/Delaunay/voir.git -b async_timer
git clone https://github.com/Delaunay/torchcompat.git
git clone https://github.com/huggingface/optimum-habana.git

wget -nv https://vault.habana.ai/artifactory/gaudi-installer/1.15.1/habanalabs-installer.sh
Expand All @@ -34,7 +33,6 @@ install_prepare() {

which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat

# Override dependencies for HPU
# milabench needs pyhlml
Expand All @@ -46,7 +44,6 @@ install_prepare() {
. $BENCHMARK_VENV/bin/activate
which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat
pip install -e $MILABENCH_WORDIR/optimum-habana

(
Expand Down
3 changes: 0 additions & 3 deletions scripts/article/run_rocm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ install_prepare() {

git clone https://github.com/mila-iqia/milabench.git -b intel
git clone https://github.com/Delaunay/voir.git -b async_timer
git clone https://github.com/Delaunay/torchcompat.git

. $MILABENCH_WORDIR/env/bin/activate
pip install -e $MILABENCH_WORDIR/milabench
Expand All @@ -31,13 +30,11 @@ install_prepare() {

which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat

(
. $BENCHMARK_VENV/bin/activate
which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
)

Expand Down
3 changes: 0 additions & 3 deletions scripts/article/run_xpu.sh
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ install_prepare() {

git clone https://github.com/mila-iqia/milabench.git -b intel
git clone https://github.com/Delaunay/voir.git -b async_timer
git clone https://github.com/Delaunay/torchcompat.git

# XPU manager is necessary
wget -nv https://github.com/intel/xpumanager/releases/download/V1.2.36/xpumanager_1.2.36_20240428.081009.377f9162.u22.04_amd64.deb
Expand All @@ -35,13 +34,11 @@ install_prepare() {

which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat

(
. $BENCHMARK_VENV/bin/activate
which pip
pip install -e $MILABENCH_WORDIR/voir
pip install -e $MILABENCH_WORDIR/torchcompat

# Override dependencies for XPU
pip install torch, torchvision torchaudio intel-extension-for-pytorch --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
Expand Down

0 comments on commit 5a13ead

Please sign in to comment.