Skip to content

Update L0_python_backend test scenario #8526

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mc-nv merged 6 commits into from
Nov 20, 2025
Merged

Update L0_python_backend test scenario #8526

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
fceb4b0
tests: L0_backend_python - to define CUDAARCHS to pick proper CUDA ar…
mc-nv Nov 19, 2025
508c69d
Variable modification, switch to 'sed' utility
mc-nv Nov 19, 2025
e099777
Including /usr/include dir for build
mc-nv Nov 19, 2025
b7386d6
Switching to newer pytorch
mc-nv Nov 20, 2025
8f7f9eb
Set pytorch version to latest one
mc-nv Nov 20, 2025
9940f37
Update pytorch version for conda environment
mc-nv Nov 20, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 16 additions & 7 deletions qa/L0_backend_python/common.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -87,11 +87,20 @@ create_python_backend_stub() {
git clone ${TRITON_REPO_ORGANIZATION}/python_backend -b $PYTHON_BACKEND_REPO_TAG
CUDA_PATH=$(readlink -f /usr/local/cuda)
export CMAKE_POLICY_VERSION_MINIMUM=3.5
(cd python_backend/ && mkdir builddir && cd builddir && \
export CMAKE_POLICY_VERSION_MINIMUM=3.5 && \
cmake -DTRITON_ENABLE_GPU=ON -DCMAKE_CUDA_COMPILER=$CUDA_PATH/bin/nvcc \
-DCUDAToolkit_ROOT=$CUDA_PATH -DTRITON_REPO_ORGANIZATION:STRING=${TRITON_REPO_ORGANIZATION} \
-DTRITON_BACKEND_REPO_TAG=$TRITON_BACKEND_REPO_TAG -DTRITON_COMMON_REPO_TAG=$TRITON_COMMON_REPO_TAG \
-DTRITON_CORE_REPO_TAG=$TRITON_CORE_REPO_TAG -DPYBIND11_PYTHON_VERSION=$PY_VERSION ../ && \
make -j18 triton-python-backend-stub)
(cd python_backend/ \
&& mkdir builddir \
&& cd builddir \
&& export CMAKE_POLICY_VERSION_MINIMUM=3.5 \
&& cmake \
-DCMAKE_CUDA_COMPILER=$CUDA_PATH/bin/nvcc \
-DCMAKE_INCLUDE_PATH:STRING=/usr/include \
-DCUDAToolkit_ROOT=$CUDA_PATH \
-DPYBIND11_PYTHON_VERSION=$PY_VERSION \
-DTRITON_BACKEND_REPO_TAG=$TRITON_BACKEND_REPO_TAG \
-DTRITON_COMMON_REPO_TAG=$TRITON_COMMON_REPO_TAG \
-DTRITON_CORE_REPO_TAG=$TRITON_CORE_REPO_TAG \
-DTRITON_ENABLE_GPU=ON \
-DTRITON_REPO_ORGANIZATION:STRING=${TRITON_REPO_ORGANIZATION} \
-S ../ \
&& cmake --build . --target triton-python-backend-stub -j18)
}
2 changes: 1 addition & 1 deletion qa/L0_backend_python/env/test.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ install_conda
path_to_conda_pack='$$TRITON_MODEL_DIRECTORY/python_3_12_environment.tar.gz'
create_conda_env "3.12" "python-3-12"
conda install -c conda-forge libstdcxx-ng=14 -y
TORCH_VERSION="2.6.0"
TORCH_VERSION="2.8.0"
conda install numpy=1.26.4 -y
if [ $TRITON_RHEL -eq 1 ]; then
TORCH_VERISON="2.17.0"
Expand Down
4 changes: 2 additions & 2 deletions qa/L0_backend_python/setup_python_enviroment.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,10 +56,10 @@ conda update -n base -c defaults conda -y
# been setup correctly.
if [ ${PYTHON_ENV_VERSION} = "11" ]; then
create_conda_env "3.11" "python-3-11"
conda install pytorch=2.6.0 -y
conda install pytorch=2.8.0 -y
conda install -c conda-forge libstdcxx-ng=14 -y
conda install numpy=1.23.5 -y
EXPECTED_VERSION_STRING="Python version is 3.11, NumPy version is 1.23.5, and PyTorch version is 2.6.0"
EXPECTED_VERSION_STRING="Python version is 3.11, NumPy version is 1.23.5, and PyTorch version is 2.8.0"
create_python_backend_stub
conda-pack -o python3.11.tar.gz
path_to_conda_pack="$PWD/python-3-11"
Expand Down
Loading