torchforge/.github/workflows/build_vllm.yaml at 89fcd8787effcc5967584f914fbce98f1a8e8983 · meta-pytorch/torchforge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
name: Build pinned vLLM against PyTorch nightly and upload

on:
  push:
    branches:
      - nightly
  workflow_dispatch:

permissions:
  id-token: write
  contents: read

jobs:
  build:
    name: forge-cu128-nightly
    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@vllm-push
    strategy:
      fail-fast: false
    with:
      repository: meta-pytorch/forge
      ref: ""
      test-infra-repository: pytorch/test-infra
      test-infra-ref: vllm-push
      run-smoke-test: false
      wheel-nightly-policy: gha_workflow_preview_build_wheels
      wheel-upload-path: whl/preview/forge/
      package-name: forge
      channel: test # Hack here to make sure stable pytorch is used
      build-matrix: |
        {
          "include": [
            {
              "python_version": "3.10",
              "gpu_arch_type": "cpu",
              "gpu_arch_version": "12.8",
              "desired_cuda": "cu128",
              "container_image": "pytorch/manylinux2_28-builder:cuda12.8",
              "package_type": "manywheel",
              "build_name": "manywheel-py3_10-cuda12_8",
              "validation_runner": "linux.12xlarge.memory",
              "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128",
              "channel": "test",
              "upload_to_base_bucket": "no",
              "stable_version": "2.9.0",
              "use_split_build": false
            },
            {
              "python_version": "3.11",
              "gpu_arch_type": "cpu",
              "gpu_arch_version": "12.8",
              "desired_cuda": "cu128",
              "container_image": "pytorch/manylinux2_28-builder:cuda12.8",
              "package_type": "manywheel",
              "build_name": "manywheel-py3_11-cuda12_8",
              "validation_runner": "linux.12xlarge.memory",
              "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128",
              "channel": "test",
              "upload_to_base_bucket": "no",
              "stable_version": "2.9.0",
              "use_split_build": false
            },
            {
              "python_version": "3.12",
              "gpu_arch_type": "cpu",
              "gpu_arch_version": "12.8",
              "desired_cuda": "cu128",
              "container_image": "pytorch/manylinux2_28-builder:cuda12.8",
              "package_type": "manywheel",
              "build_name": "manywheel-py3_12-cuda12_8",
              "validation_runner": "linux.12xlarge.memory",
              "installation": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128",
              "channel": "test",
              "upload_to_base_bucket": "no",
              "stable_version": "2.9.0",
              "use_split_build": false
            },
          ]
        }
      pre-script: .github/packaging/pre_build_cpu.sh
      post-script: .github/packaging/post_build_script.sh
      trigger-event: ${{ github.event_name }}
      build-platform: 'python-build-package'