1 files changed, 24 insertions, 13 deletions
diff --git a/ldcg-python-manylinux-tensorflow/ansible/roles/tensorflow/tasks/loop.yml b/ldcg-python-manylinux-tensorflow/ansible/roles/tensorflow/tasks/loop.yml
index da2eeef1df..e9ba473675 100644
--- a/ldcg-python-manylinux-tensorflow/ansible/roles/tensorflow/tasks/loop.yml
+++ b/ldcg-python-manylinux-tensorflow/ansible/roles/tensorflow/tasks/loop.yml
@@ -1,19 +1,30 @@
-- name: set TensorFlow version
-  set_fact:
+---
+- name: Set TensorFlow and other package versions
+  ansible.builtin.set_fact:
     tensorflow_version: "{{ tf_ver.value.tensorflow_version }}"
-    numpy_version: "{{ tf_ver.value.numpy_version }}"
-    h5py_version: "{{ tf_ver.value.h5py_version }}"
-
-- name: build Bazel
-  include_role:
-    name: bazel
-  vars:
     bazel_version: "{{ tf_ver.value.bazel_version }}"
+    onednn_ut_exclude: '{{ "-" + tf_ver.value.onednn_ut_exclude | join(" -") if onednn_enable_opt and tf_ver.value.onednn_ut_exclude else "" }}'
+    clang_ut_exclude: '{{ "-" + tf_ver.value.unit_tests_clang_exclude | join(" -") if clang_build and tf_ver.value.unit_tests_clang_exclude else "" }}'
+    gcc_ut_exclude: '{{ "-" + tf_ver.value.unit_tests_gcc_exclude | join(" -") if not clang_build and tf_ver.value.unit_tests_gcc_exclude else "" }}'
+    image_prefix: "{{ tf_ver.value.image_prefix }}"
+
+- name: Set build dir for TensorFlow{{ variant_name }} {{ tensorflow_version }}
+  ansible.builtin.set_fact:
+    tensorflow_build_dir: "{{ build_dir }}/tensorflow{{ variant_name }}-{{ tensorflow_version }}"
+
+- name: Remove TensorFlow{{ variant_name }} {{ tensorflow_version }} sources from previous build
+  ansible.builtin.file:
+    path: "{{ tensorflow_build_dir }}/"
+    state: absent
+
+- name: Fetch TensorFlow
+  ansible.builtin.include_tasks:
+    file: fetch.yml
 
-- name: go through each Python version
-  include_role:
-    name: tensorflow
-    tasks_from: manylinux-build.yml
+- name: Build TensorFlow for each Python version (dual abi)
+  ansible.builtin.include_role:
+    name: tensorflow_in_container
+    tasks_from: dual-abi-build.yml
   loop: "{{ tf_ver.value.python_versions }}"
   loop_control:
     loop_var: py_ver