diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml index d1dde2aeb..8c97a9f3b 100644 --- a/.github/workflows/wheels_build.yml +++ b/.github/workflows/wheels_build.yml @@ -65,6 +65,13 @@ jobs: submodules: recursive path: "." fetch-depth: 0 # for tags + + - name: HACKFIX for cutlass compiler bug + if: runner.os == 'Windows' + run: + # See https://github.com/NVIDIA/cutlass/issues/1732 + rm -f third_party/cutlass/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp + touch third_party/cutlass/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp - name: Setup Runner uses: ./.github/actions/setup-build-cuda with: diff --git a/.github/workflows/win-build.yml b/.github/workflows/win-build.yml index d68083606..02a32d497 100644 --- a/.github/workflows/win-build.yml +++ b/.github/workflows/win-build.yml @@ -51,6 +51,13 @@ jobs: python -c "import tempfile; print(tempfile.gettempdir())" python -c "import tempfile; assert(len(tempfile.gettempdir()) < 30)" + - name: HACKFIX for cutlass compiler bug + if: runner.os == 'Windows' + run: + # See https://github.com/NVIDIA/cutlass/issues/1732 + rm -f third_party/cutlass/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp + touch third_party/cutlass/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp + - name: Setup Runner uses: ./.github/actions/setup-build-cuda with: @@ -73,9 +80,6 @@ jobs: $PY -c "import torch; print('torch.cuda', torch.version.cuda)" ninja --version - - name: Work around long-path issue - run: | - - name: Create sdist run: $PY setup.py sdist @@ -90,6 +94,6 @@ jobs: XFORMERS_MORE_DETAILS=1 $PY -m xformers.info # Open an SSH session on failure to debug - - name: Setup tmate session - if: ${{ failure() }} - uses: mxschmitt/action-tmate@v3 \ No newline at end of file + # - name: Setup tmate session + # if: ${{ failure() }} + # uses: mxschmitt/action-tmate@v3 \ No newline at end of file