chore: improve develop.sh and install.sh with better error handling

[yurekami]

Summary

Improve develop.sh and install.sh scripts with better error handling, prerequisite checks, and user feedback.

Changes

Both Scripts

• Bash strict mode: Added set -euo pipefail for better error detection
• Colored output: Green for success, yellow for warnings, red for errors
• Prerequisite checks:
  • Verify Python is available
  • Verify pip is available (install.sh only)
  • Check CUDA_HOME or fallback to /usr/local/cuda
• Automatic submodule init: Initialize submodules if missing
• Proper cleanup: Use trap to ensure directory is restored on exit
• Success messages: Show helpful next steps after completion

develop.sh Improvements

# Before
python setup.py build
# After
echo "Building DeepGEMM..."
python setup.py build
echo -e "${GREEN}Development environment setup complete!${NC}"
echo "You can now run tests with:"
echo "  python tests/test_core.py"

install.sh Improvements

# Before
pip install dist/*.whl --force-reinstall
# After
wheel_file=$(find dist -name "*.whl" -type f | head -n 1)
pip install "$wheel_file" --force-reinstall
echo "You can verify the installation with:"
echo "  python -c \"import deep_gemm; print(deep_gemm.__path__)\""

Benefits

• Better debugging: Script fails fast on errors with meaningful messages
• Smoother onboarding: New contributors get clear feedback
• Robustness: Handles missing prerequisites gracefully
• Consistency: Both scripts follow the same patterns

🤖 Generated with Claude Code