From 59a2d633938d61100fb50caa345a3183abfe458d Mon Sep 17 00:00:00 2001 From: Steboss Date: Mon, 3 Nov 2025 11:16:51 +0000 Subject: [PATCH 1/5] try to see if we can buld this fine --- .github/container/pip-finalize.sh | 2 ++ .github/workflows/ci.yaml | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/container/pip-finalize.sh b/.github/container/pip-finalize.sh index 6e2a59aed..a9ffe1ece 100755 --- a/.github/container/pip-finalize.sh +++ b/.github/container/pip-finalize.sh @@ -53,6 +53,8 @@ else # Replace any tensorflow==X with tensorflow-cpu==X in requirements.txt only on amd64 if [[ "$(uname -m)" = "x86_64" ]]; then sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt + elif [[ "$(uname -m)" = "arm64" ]]; then + sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-metal==1.1.0/' requirements.txt else echo "Skipping TF on $(uname -m)" fi diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 1031f1e25..c411a046f 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -222,7 +222,7 @@ jobs: CUDA_IMAGE: ${{ needs.metadata.outputs.CUDA_IMAGE }} MANIFEST_ARTIFACT_NAME: ${{ needs.metadata.outputs.MANIFEST_ARTIFACT_NAME }} SOURCE_URLREFS: ${{ needs.bump-manifest.outputs.SOURCE_URLREFS }} - MODE: ${{ github.event_name == 'workflow_dispatch' && inputs.MODE || 'full' }} + MODE: "axlearn" # ${{ github.event_name == 'workflow_dispatch' && inputs.MODE || 'full' }} secrets: inherit # Only merge if everything succeeds From 9abce01458427e733e2c661576a2d1176ecf1977 Mon Sep 17 00:00:00 2001 From: Steboss Date: Mon, 3 Nov 2025 13:37:56 +0000 Subject: [PATCH 2/5] modify with tensorflow-metal --- .github/container/pip-finalize.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/container/pip-finalize.sh b/.github/container/pip-finalize.sh index a9ffe1ece..79dd1124d 100755 --- a/.github/container/pip-finalize.sh +++ b/.github/container/pip-finalize.sh @@ -53,8 +53,8 @@ else # Replace any tensorflow==X with tensorflow-cpu==X in requirements.txt only on amd64 if [[ "$(uname -m)" = "x86_64" ]]; then sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt - elif [[ "$(uname -m)" = "arm64" ]]; then - sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-metal==1.1.0/' requirements.txt + elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then + sed -i 's/^tensorflow==.*$/tensorflow-metal==1.1.0/' requirements.txt else echo "Skipping TF on $(uname -m)" fi From e4b0e512cf68ebd5d894cbf16669ac8944b354cd Mon Sep 17 00:00:00 2001 From: Steboss Date: Mon, 3 Nov 2025 15:04:46 +0000 Subject: [PATCH 3/5] try a standard version --- .github/container/pip-finalize.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/container/pip-finalize.sh b/.github/container/pip-finalize.sh index 79dd1124d..2fdd759ba 100755 --- a/.github/container/pip-finalize.sh +++ b/.github/container/pip-finalize.sh @@ -54,7 +54,7 @@ else if [[ "$(uname -m)" = "x86_64" ]]; then sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then - sed -i 's/^tensorflow==.*$/tensorflow-metal==1.1.0/' requirements.txt + sed -i 's/^tensorflow==.*$/tensorflow-metal/' requirements.txt else echo "Skipping TF on $(uname -m)" fi From e37b967b8231731b29e41f222b8bc7ee7aab24c3 Mon Sep 17 00:00:00 2001 From: Steboss Date: Mon, 3 Nov 2025 15:41:22 +0000 Subject: [PATCH 4/5] test with tensorflow aarch 64 --- .github/container/pip-finalize.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/container/pip-finalize.sh b/.github/container/pip-finalize.sh index 2fdd759ba..f61da9d6a 100755 --- a/.github/container/pip-finalize.sh +++ b/.github/container/pip-finalize.sh @@ -54,7 +54,7 @@ else if [[ "$(uname -m)" = "x86_64" ]]; then sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then - sed -i 's/^tensorflow==.*$/tensorflow-metal/' requirements.txt + sed -i 's/^tensorflow==.*$/tensorflow-aarch64/' requirements.txt else echo "Skipping TF on $(uname -m)" fi From eaada3bff88312aafb77dfef35099885de26152e Mon Sep 17 00:00:00 2001 From: Steboss Date: Tue, 4 Nov 2025 12:11:45 +0000 Subject: [PATCH 5/5] make sure we re installing tensorflow 2181 --- .github/container/pip-finalize.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/container/pip-finalize.sh b/.github/container/pip-finalize.sh index f61da9d6a..4c0122c6e 100755 --- a/.github/container/pip-finalize.sh +++ b/.github/container/pip-finalize.sh @@ -54,7 +54,7 @@ else if [[ "$(uname -m)" = "x86_64" ]]; then sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then - sed -i 's/^tensorflow==.*$/tensorflow-aarch64/' requirements.txt + sed -i 's/^tensorflow==.*$/tensorflow==2.18.1/' requirements.txt else echo "Skipping TF on $(uname -m)" fi