Skip to content

Commit 0402227

Browse files
ddunlGoogle-ML-Automation
authored andcommitted
Move TensorFlow CPU/GPU builds to presubmit
PiperOrigin-RevId: 726214726
1 parent 3923508 commit 0402227

File tree

2 files changed

+15
-16
lines changed

2 files changed

+15
-16
lines changed

.github/workflows/ci.yml

+14
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,20 @@ jobs:
6767
pretty_name: "JAX Linux x86 GPU T4",
6868
repo: "jax-ml/jax",
6969
},
70+
{
71+
pool: "linux-x86-n2-16",
72+
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest",
73+
kokoro_job_name: "tensorflow-linux-x86-cpu",
74+
pretty_name: "TensorFlow Linux x86 CPU",
75+
repo: "tensorflow/tensorflow",
76+
},
77+
{
78+
pool: "linux-x86-g2-16-l4-1gpu",
79+
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest",
80+
kokoro_job_name: "tensorflow-linux-x86-gpu-t4",
81+
pretty_name: "TensorFlow Linux x86 GPU T4",
82+
repo: "tensorflow/tensorflow",
83+
},
7084
]
7185
name: ${{ matrix.job_info.pretty_name }}
7286
runs-on: ${{ matrix.job_info.pool }}

.github/workflows/ci_continuous_only.yml

+1-16
Original file line numberDiff line numberDiff line change
@@ -33,22 +33,7 @@ jobs:
3333
# Don't fail fast - want to see results for all builds even if one fails.
3434
fail-fast: false
3535
matrix:
36-
job_info: [
37-
{
38-
pool: "linux-x86-n2-16",
39-
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest",
40-
kokoro_job_name: "tensorflow-linux-x86-cpu",
41-
pretty_name: "TensorFlow Linux x86 CPU",
42-
repo: "tensorflow/tensorflow",
43-
},
44-
{
45-
pool: "linux-x86-g2-16-l4-1gpu",
46-
container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest",
47-
kokoro_job_name: "tensorflow-linux-x86-gpu-t4",
48-
pretty_name: "TensorFlow Linux x86 GPU T4",
49-
repo: "tensorflow/tensorflow",
50-
},
51-
]
36+
job_info: []
5237
name: ${{ matrix.job_info.pretty_name }}
5338
runs-on: ${{ matrix.job_info.pool }}
5439
container: ${{ matrix.job_info.container }}

0 commit comments

Comments
 (0)