update DS_BUILD_EVOFORMER_ATTN=0

shwj114514 · Mar 28, 2024 · 31c6e97 · 31c6e97
1 parent 64c1595
commit 31c6e97
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 79 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -97,7 +97,7 @@ jobs:
             (yum install -y libaio-devel libffi-devel || apt-get install -y libaio-dev libffi-devel || apk add --update --no-cache libaio-devel libffi-devel || true) &&
             (yum install -y libc6 || apt-get install -y libc6 || apk add --update --no-cache libc6 || true) &&
             bash build_scripts/prepare_for_build.sh
-          CIBW_ENVIRONMENT: CFLAGS="-I/usr/include" LDFLAGS="-L/usr/lib -laio" AUDITWHEEL_PLAT="manylinux_2_24_$(uname -m)" MATRIX_CUDA_VERSION=${{ env.MATRIX_CUDA_VERSION }} MATRIX_TORCH_VERSION=${{ env.MATRIX_TORCH_VERSION }} MATRIX_PYTHON_VERSION=${{ env.MATRIX_PYTHON_VERSION }} CI_CUDA_VERSION=${{ env.CI_CUDA_VERSION }} CI_TORCH_VERSION=${{ env.CI_TORCH_VERSION }} CI_PYTHON_VERSION=${{ env.CI_PYTHON_VERSION }} CUTLASS_PATH=/project/cutlass TORCH_CUDA_ARCH_LIST="6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX" DS_BUILD_STRING="+cu${{ env.MATRIX_CUDA_VERSION }}torch${{ env.MATRIX_TORCH_VERSION }}" DS_BUILD_OPS=1 DS_BUILD_SPARSE_ATTN=0
+          CIBW_ENVIRONMENT: CFLAGS="-I/usr/include" LDFLAGS="-L/usr/lib -laio" AUDITWHEEL_PLAT="manylinux_2_24_$(uname -m)" MATRIX_CUDA_VERSION=${{ env.MATRIX_CUDA_VERSION }} MATRIX_TORCH_VERSION=${{ env.MATRIX_TORCH_VERSION }} MATRIX_PYTHON_VERSION=${{ env.MATRIX_PYTHON_VERSION }} CI_CUDA_VERSION=${{ env.CI_CUDA_VERSION }} CI_TORCH_VERSION=${{ env.CI_TORCH_VERSION }} CI_PYTHON_VERSION=${{ env.CI_PYTHON_VERSION }} CUTLASS_PATH=/project/cutlass TORCH_CUDA_ARCH_LIST="3.5;3.7;5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX" DS_BUILD_STRING="+cu${{ env.MATRIX_CUDA_VERSION }}torch${{ env.MATRIX_TORCH_VERSION }}" DS_BUILD_OPS=1 DS_BUILD_SPARSE_ATTN=0 DS_BUILD_EVOFORMER_ATTN=0
           CIBW_SKIP: "*musllinux*"
           CIBW_REPAIR_WHEEL_COMMAND_LINUX: "auditwheel repair -w /tmp/cibuildwheel/repaired_wheel {wheel} --exclude libc10.so --exclude libc10_cuda.so --exclude libtorch.so --exclude libtorch_python.so --exclude libtorch_cpu.so --exclude libtorch_cuda.so --exclude libtorch_cuda_cpp.so --exclude libtorch_cuda_cu.so"
 

diff --git a/patches/evoformer_atten_v0.14.0.py b/patches/evoformer_atten_v0.14.0.py
diff --git a/prepare_for_build.sh b/prepare_for_build.sh
@@ -1,7 +1,3 @@
-# patch
-mv /project/build_scripts/patches/evoformer_atten_v0.14.0.py /project/op_builder/evoformer_attn.py 
-
-
 pip install setuptools==68.0.0 wheel auditwheel-symbols
 pip install lit
 # We want to figure out the CUDA version to download pytorch