Skip to content

[BE] Fix: Show actual error messages in Triton installation script #49

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

Closed
FindHao wants to merge 4 commits into from
Closed

[BE] Fix: Show actual error messages in Triton installation script #49

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
9f6ae50
Improve Triton installation verification and error reporting
FindHao Jul 31, 2025
e74d871
Enhance Triton installation verification with detailed diagnostics
FindHao Jul 31, 2025
676d726
Enhance Triton installation verification with basic Python functional…
FindHao Jul 31, 2025
75144a7
Improve Triton installation scripts with enhanced import verification…
FindHao Aug 1, 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
18 changes: 17 additions & 1 deletion .ci/install-triton-pip.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,12 +40,28 @@ show_elapsed

# Verify Triton installation
echo "Verifying Triton installation..."
if python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>/dev/null; then

set +e # Temporarily disable exit on error
IMPORT_OUTPUT=$(python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>&1)
IMPORT_EXITCODE=$?
set -e # Re-enable exit on error

echo "Import exit code: $IMPORT_EXITCODE"
echo "Import output: $IMPORT_OUTPUT"

if [ $IMPORT_EXITCODE -eq 0 ]; then
echo "$IMPORT_OUTPUT"
python -c "import triton; print(f'Triton path: {triton.__file__}')"
echo "✅ Triton installation verified successfully"
show_elapsed
echo "🎉 Triton installation completed successfully!"
else
echo "❌ ERROR: Failed to import Triton"
echo "Import error details:"
echo "$IMPORT_OUTPUT"
echo ""
echo "Additional diagnostic information:"
echo "Installed packages containing 'triton':"
pip list | grep -i triton || echo "No triton packages found"
exit 1
fi
42 changes: 34 additions & 8 deletions .ci/install-triton.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -72,9 +72,11 @@ fi
# ImportError: /opt/miniconda3/envs/tritonparse/bin/../lib/libstdc++.so.6:
# version `GLIBCXX_3.4.30' not found (required by /tmp/triton/python/triton/_C/libtriton.so)
echo "Updating libstdc++ to match system version..."
conda install -y -c conda-forge libstdcxx-ng=12.3.0
# Use the latest version for Ubuntu 22.04 that includes GLIBCXX_3.4.32
conda install -y -c conda-forge libstdcxx-ng=15.1.0
# Check if the update was successful
strings /opt/miniconda3/envs/tritonparse/lib/libstdc++.so.6 | grep GLIBCXX | tail -5
echo "Checking libstdc++ version after update:"
strings /opt/miniconda3/envs/tritonparse/lib/libstdc++.so.6 | grep GLIBCXX | tail -10

# Uninstall existing pytorch-triton
echo "Uninstalling existing pytorch-triton..."
Expand Down Expand Up @@ -143,7 +145,23 @@ show_elapsed

# Verify Triton installation
echo "Verifying Triton installation..."
if python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>/dev/null; then
echo "Python path: $(which python)"
echo "Python version: $(python --version)"
echo "PYTHONPATH: $PYTHONPATH"
echo "Testing basic Python functionality..."
python -c "print('Python works')" || echo "❌ Basic Python test failed"
echo "Attempting to import triton..."

set +e # Temporarily disable exit on error
IMPORT_OUTPUT=$(python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>&1)
IMPORT_EXITCODE=$?
set -e # Re-enable exit on error

echo "Import exit code: $IMPORT_EXITCODE"
echo "Import output: $IMPORT_OUTPUT"

if [ $IMPORT_EXITCODE -eq 0 ]; then
echo "$IMPORT_OUTPUT"
python -c "import triton; print(f'Triton path: {triton.__file__}')"
echo "✅ Triton installation verified successfully"

Expand All @@ -155,11 +173,19 @@ if python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>/
echo "🎉 Triton installation completed successfully!"
else
echo "❌ ERROR: Failed to import triton"
echo "This might be due to libstdc++ version issues"
echo "Checking system libstdc++ version:"
strings /usr/lib/x86_64-linux-gnu/libstdc++.so.6 | grep GLIBCXX | tail -5 || echo "Could not check system libstdc++"
echo "Checking conda libstdc++ version:"
strings /opt/miniconda3/envs/tritonparse/lib/libstdc++.so.6 | grep GLIBCXX | tail -5 || echo "Could not check conda libstdc++"
echo "Import error details:"
echo "$IMPORT_OUTPUT"
echo ""
echo "Additional diagnostic information:"
echo "Installed packages containing 'triton':"
pip list | grep -i triton || echo "No triton packages found"
echo ""
echo "Python sys.path:"
python -c "import sys; print('\n'.join(sys.path))"
echo ""
echo "Checking if triton directory exists in site-packages:"
python -c "import site; print([p for p in site.getsitepackages()])" 2>/dev/null || echo "Could not get site-packages"
find $(python -c "import site; print(' '.join(site.getsitepackages()))" 2>/dev/null) -name "*triton*" 2>/dev/null || echo "Could not find triton in site-packages"

# Clean up cache on failure to prevent corruption
echo "🧹 Cleaning up cache due to installation failure..."
Expand Down