helion/.github/matrix.json at 6bc95245b886b9fd2c9b3220bdb72963f525e284 · pytorch/helion · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
{
  "include": [
    {
      "runner": "linux.g5.4xlarge.nvidia.gpu",
      "python-version": "3.10",
      "ref-eager": false,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-2.11",
      "alias": "a10g",
      "backend": "triton"
    },
    {
      "runner": "linux.g5.4xlarge.nvidia.gpu",
      "python-version": "3.14",
      "ref-eager": false,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-nightly",
      "alias": "a10g",
      "backend": "triton"
    },
    {
      "runner": "linux.g5.4xlarge.nvidia.gpu",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-2.11",
      "alias": "a10g-dtype-asserts",
      "dtype-asserts": true,
      "expecttest-accept": true,
      "backend": "triton"
    },
    {
      "runner": "linux.g5.4xlarge.nvidia.gpu",
      "python-version": "3.12",
      "ref-eager": true,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-nightly",
      "alias": "a10g-ref-eager",
      "backend": "triton"
    },
    {
      "runner": "linux.aws.h100",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-nightly",
      "alias": "h100",
      "backend": "triton"
    },
    {
      "runner": "linux.aws.h100.4",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
      "runtime-version": "cu128",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-nightly",
      "alias": "h100-distributed",
      "backend": "triton"
    },
    {
      "runner": "linux.dgx.b200",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:13.1.0-devel-ubuntu24.04",
      "runtime-version": "cu130",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-2.11",
      "alias": "b200",
      "backend": "triton"
    },
    {
      "runner": "linux.dgx.b200",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:13.1.0-devel-ubuntu24.04",
      "runtime-version": "cu130",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-2.11",
      "alias": "b200",
      "backend": "tileir"
    },
    {
      "runner": "linux.rocm.gpu.gfx942.1",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "rocm/dev-ubuntu-24.04:7.0-complete",
      "runtime-version": "rocm7.0",
      "container-options": "--device=/dev/kfd --device=/dev/dri",
      "pytorch-version": "pytorch-nightly",
      "alias": "mi325x",
      "backend": "triton"
    },
    {
      "runner": "linux.google.tpuv7x.1",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "",
      "runtime-version": "tpu",
      "container-options": "",
      "pytorch-version": "pytorch-nightly",
      "alias": "tpu",
      "backend": "pallas"
    },
    {
      "runner": "linux.dgx.b200",
      "python-version": "3.12",
      "ref-eager": false,
      "image": "nvidia/cuda:13.1.0-devel-ubuntu24.04",
      "runtime-version": "cu130",
      "container-options": "--gpus all",
      "pytorch-version": "pytorch-nightly",
      "alias": "b200",
      "backend": "cute"
    }
  ]
}