Modify build script and setup.py for wheel building

Author: shuaills

sgl-kernel/build.sh

@@ -11,11 +11,11 @@ else
 fi
 
 docker run --rm \
-    -v \"$(pwd)\":/sgl-kernel \
+    -v $(pwd):/sgl-kernel \
     pytorch/manylinux-builder:cuda${CUDA_VERSION} \
-    bash -c \"
+    bash -c "
     ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \
-    ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja setuptools>=75.0 wheel numpy && \
+    ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja setuptools==75.0.0 wheel==0.41.0 numpy && \
     ${PYTHON_ROOT_PATH}/bin/python -c 'import setuptools; import sys; print(\"Setuptools path:\", setuptools.__file__); print(\"Python path priority:\", sys.path)' && \
     export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
     export CUDA_VERSION=${CUDA_VERSION} && \
@@ -25,5 +25,7 @@ docker run --rm \
     mkdir -p /usr/lib/x86_64-linux-gnu/ && \
     ln -s /usr/local/cuda-${CUDA_VERSION}/targets/x86_64-linux/lib/stubs/libcuda.so /usr/lib/x86_64-linux-gnu/libcuda.so && \
     cd /sgl-kernel && \
-    PYTHONPATH=${PYTHON_ROOT_PATH}/lib/python${PYTHON_VERSION}/site-packages ${PYTHON_ROOT_PATH}/bin/python setup.py bdist_wheel
-    \"
+    ${PYTHON_ROOT_PATH}/bin/python -c 'import setuptools; import importlib.util; print(\"setuptools version:\", setuptools.__version__); import wheel; print(\"wheel version:\", wheel.__version__); print(\"wheel modules:\", dir(wheel))' && \
+    ls -la ${PYTHON_ROOT_PATH}/lib/python${PYTHON_VERSION}/site-packages/wheel/ && \
+    PYTHONPATH=${PYTHON_ROOT_PATH}/lib/python${PYTHON_VERSION}/site-packages ${PYTHON_ROOT_PATH}/bin/python -m pip wheel . --no-deps
+    "

sgl-kernel/setup.py

@@ -20,7 +20,7 @@
 
 import torch
 from setuptools import find_packages, setup
-from setuptools.command.bdist_wheel import bdist_wheel
+from wheel.bdist_wheel import bdist_wheel
 from setuptools.command.build_py import build_py
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension