Skip to content

Commit

Permalink
wheel-4
Browse files Browse the repository at this point in the history
  • Loading branch information
abcdabcd987 committed Dec 22, 2023
1 parent 64bdea8 commit 165e548
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 1 deletion.
1 change: 1 addition & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
include version.txt
include src/punica/_build_meta.py
recursive-include csrc *
recursive-include third_party/cutlass/include *
recursive-include third_party/flashinfer/include *
Expand Down
7 changes: 6 additions & 1 deletion ci/run-ci-build-wheel.bash
Original file line number Diff line number Diff line change
Expand Up @@ -23,16 +23,21 @@ mkdir -p $HOME
CUDA_MAJOR="${PUNICA_CI_CUDA_VERSION%.*}"
CUDA_MINOR="${PUNICA_CI_CUDA_VERSION#*.}"
PYVER="${PUNICA_CI_PYTHON_VERSION//./}"
export PATH="$HOME/.local/bin:$PATH"
export PATH="/opt/python/cp${PYVER}-cp${PYVER}/bin:$PATH"


echo "::group::Install PyTorch"
pip install torch==$PUNICA_CI_TORCH_VERSION --index-url "https://download.pytorch.org/whl/cu${CUDA_MAJOR}${CUDA_MINOR}"
echo "::endgroup::"

echo "::group::Install build system"
pip install ninja numpy
pip install --upgrade setuptools wheel build
echo "::endgroup::"


echo "::group::Build wheel for Punica"
cd "$PROJECT_ROOT"
pip install ninja numpy build
python -m build --no-isolation
echo "::endgroup::"

0 comments on commit 165e548

Please sign in to comment.