sync to upstream

Author: nitins17

.github/workflows/asan.yaml

@@ -21,38 +21,35 @@ jobs:
       run:
         shell: bash -l {0}
     steps:
-      # # Install git before actions/checkout as otherwise it will download the code with the GitHub
-      # # REST API and therefore any subsequent git commands will fail.
-      # - name: Install clang 18
-      #   env:
-      #     DEBIAN_FRONTEND: noninteractive
-      #   run: |
-      #     apt update
-      #     apt install -y clang-18 libstdc++-14-dev build-essential libssl-dev \
-      #       zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev curl git \
-      #       libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev \
-      #       libffi-dev liblzma-dev
-      # - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
-      #   with:
-      #     path: jax
-      # - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
-      #   with:
-      #     repository: python/cpython
-      #     path: cpython
-      #     ref: v3.13.0
-      # - name: Build CPython with ASAN enabled
-      #   env:
-      #     ASAN_OPTIONS: detect_leaks=0
-      #   run: |
-      #     cd cpython
-      #     mkdir ${GITHUB_WORKSPACE}/cpythonasan
-      #     CC=clang-18 CXX=clang++-18 ./configure --prefix ${GITHUB_WORKSPACE}/cpythonasan --with-address-sanitizer --without-pymalloc
-      #     make -j64
-      #     make install
-      #     ${GITHUB_WORKSPACE}/cpythonasan/bin/python3 -m venv ${GITHUB_WORKSPACE}/venv
-      # Halt for testing
-      - name: Wait For Connection
-        uses: google-ml-infra/actions/ci_connection@main
+      # Install git before actions/checkout as otherwise it will download the code with the GitHub
+      # REST API and therefore any subsequent git commands will fail.
+      - name: Install clang 18
+        env:
+          DEBIAN_FRONTEND: noninteractive
+        run: |
+          apt update
+          apt install -y clang-18 libstdc++-14-dev build-essential libssl-dev \
+            zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev curl git \
+            libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev \
+            libffi-dev liblzma-dev
+      - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        with:
+          path: jax
+      - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        with:
+          repository: python/cpython
+          path: cpython
+          ref: v3.13.0
+      - name: Build CPython with ASAN enabled
+        env:
+          ASAN_OPTIONS: detect_leaks=0
+        run: |
+          cd cpython
+          mkdir ${GITHUB_WORKSPACE}/cpythonasan
+          CC=clang-18 CXX=clang++-18 ./configure --prefix ${GITHUB_WORKSPACE}/cpythonasan --with-address-sanitizer --without-pymalloc
+          make -j64
+          make install
+          ${GITHUB_WORKSPACE}/cpythonasan/bin/python3 -m venv ${GITHUB_WORKSPACE}/venv
       - name: Install JAX test requirements
         env:
           ASAN_OPTIONS: detect_leaks=0