From a69380acdd6ce8180133d6d975cc6dc88192478d Mon Sep 17 00:00:00 2001
From: Tongze Wang <wtongze@gmail.com>
Date: Thu, 12 Sep 2024 13:13:01 -0700
Subject: [PATCH] feat: build files

---
 pytorch.dockerfile | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pytorch.dockerfile b/pytorch.dockerfile
index 22f9fbd..875f91c 100644
--- a/pytorch.dockerfile
+++ b/pytorch.dockerfile
@@ -48,7 +48,6 @@ PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") python${PYTHON_VER
 # https://github.com/pytorch/pytorch/issues/125230
 # https://github.com/AngryLoki/pytorch/commit/38d6f115c65b67946de6765165ece0288c8416d4
 
-# Test
 RUN <<EOF
 sudo apt-get update
 sudo apt-get install -y ffmpeg
@@ -73,7 +72,13 @@ WORKDIR /home/rocm/audio
 RUN USE_ROCM=ON \
 PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") \
 python${PYTHON_VERSION} setup.py bdist_wheel
-# Test End
+
+RUN <<EOF
+mkdir /home/rocm/build
+cp /home/rocm/pytorch/dist/*.whl /home/rocm/build
+cp /home/rocm/vision/dist/*.whl /home/rocm/build
+cp /home/rocm/audio/dist/*.whl /home/rocm/build
+EOF
 
 WORKDIR /home/rocm
 CMD [ "bash", "-l" ]