diff --git a/.github/workflows/H-Coverage.yml b/.github/workflows/H-Coverage.yml
index 49f9efa3ba31fc..766fd8092fb9fa 100644
--- a/.github/workflows/H-Coverage.yml
+++ b/.github/workflows/H-Coverage.yml
@@ -104,7 +104,7 @@ jobs:
           LITE_GIT_TAG: develop
           WITH_UNITY_BUILD: "ON"
           WITH_FA_BUILD_WITH_CACHE: "ON"
-          PY_VERSION: "3.10"
+          PY_VERSION: "3.12"
           INFERENCE_DEMO_INSTALL_DIR: /root/.cache/coverage
           CCACHE_MAXSIZE: 50G
           CCACHE_LIMIT_MULTIPLE: 0.8
@@ -113,7 +113,7 @@ jobs:
         run: |
           container_name=${TASK}-build-$(date +%Y%m%d-%H%M%S)
           echo "container_name=${container_name}" >> ${{ github.env }}
-          docker_image=ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:cuda129-coverage-test
+          docker_image=ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-py312-dev
           docker run -d -t --name ${container_name} \
             -v "/home/data/cfs:/home/data/cfs" \
             -v "/home/data/cfs/.cache:/root/.cache" \
@@ -197,7 +197,7 @@ jobs:
           pip install -r python/requirements.txt
           mkdir -p build && cd build
           ccache -z
-          cmake .. -DPY_VERSION=3.10 -DWITH_GPU=ON -DWITH_DISTRIBUTE=ON -DWITH_TESTING=ON -DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN="90" -DFA_JOB_POOLS_COMPILE=1 -DWITH_CUDNN_FRONTEND=ON -DON_INFER=OFF -DWITH_NVSHMEM=ON
+          cmake .. -DPY_VERSION=3.12 -DWITH_GPU=ON -DWITH_DISTRIBUTE=ON -DWITH_TESTING=ON -DCUDA_ARCH_NAME=Manual -DCUDA_ARCH_BIN="90" -DFA_JOB_POOLS_COMPILE=1 -DWITH_CUDNN_FRONTEND=ON -DON_INFER=OFF -DWITH_NVSHMEM=ON
           make -j20
           EXIT_CODE=$?
           ccache -s
@@ -223,7 +223,7 @@ jobs:
         env:
           home_path: ${{ github.workspace }}/..
           bos_file: ${{ github.workspace }}/../bos_retry/BosClient.py
-          paddle_whl: paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl
+          paddle_whl: paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl
         run: |
           docker exec -t ${{ env.container_name }} /bin/bash -c '
           echo "::group::Install bce-python-sdk"
@@ -238,7 +238,7 @@ jobs:
           fi
           cd /paddle
           mv /Paddle.tar.gz .
-          cp ./build/python/dist/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl .
+          cp ./build/python/dist/paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl .
           echo "Uploading Paddle.tar.gz"
           python ${{ env.bos_file }} Paddle.tar.gz paddle-github-action/PR/h-coverage/${{ env.PR_ID }}/${{ env.COMMIT_ID }}
           echo "Uploading coverage wheel"
@@ -283,7 +283,7 @@ jobs:
           PADDLE_VERSION: 0.0.0
           WITH_DISTRIBUTE: "ON"
           WITH_UNITY_BUILD: "ON"
-          PY_VERSION: "3.10"
+          PY_VERSION: "3.12"
           WITH_SHARED_PHI: "ON"
           GPU_DEVICES: ${{ env.GPU_DEVICES }}
           WITH_CINN: "ON"
@@ -298,7 +298,7 @@ jobs:
         run: |
           container_name=${TASK}-$(date +%Y%m%d-%H%M%S)
           echo "container_name=${container_name}" >> ${{ github.env }}
-          docker_image=ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:cuda129-coverage-test
+          docker_image=ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-py312-dev
           docker run -d -t --gpus "\"device=${GPU_DEVICES}\"" --name ${container_name} \
             -v "/home/data/cfs:/home/data/cfs" \
             -v "/home/data/cfs/.cache:/root/.cache" \
@@ -369,6 +369,7 @@ jobs:
         run: |
           docker exec -t ${{ env.container_name }} /bin/bash -c '
           source ${{ github.workspace }}/../../../proxy
+          python -m pip install pytest einops
           cd test/test_flashmask_ci
           bash run.sh
           '
@@ -391,7 +392,8 @@ jobs:
     timeout-minutes: 60
     env:
       TASK: fleet-ci-paddle-build-whl-${{ github.event.pull_request.number }}
-      docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-dev"
+      docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-py312-dev"
+      PY_VERSION: "3.12"
     steps:
       - name: Check docker image and run container
         env:
@@ -425,6 +427,7 @@ jobs:
             -e CCACHE_LIMIT_MULTIPLE \
             -e CCACHE_STATSLOG \
             -e CCACHE_SLOPPINESS \
+            -e PY_VERSION \
             -w /paddle --network host ${docker_image}
 
       - name: uv build whl
@@ -458,9 +461,9 @@ jobs:
           export LD_LIBRARY_PATH=/usr/local/cuda-12.9/compat:$LD_LIBRARY_PATH
           export IS_NVIDIA=True
           echo "Downloading Paddle.tar.gz from cfs"
-          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --no-check-certificate
+          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --no-check-certificate
           pip uninstall paddlepaddle-gpu -y
-          pip install paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
+          pip install paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
           pip install paddle-nvidia-nvshmem-cu12 --extra-index-url https://www.paddlepaddle.org.cn/packages/nightly/cu129/
           uv build --wheel --no-build-isolation -v --python $(which python)
           '
@@ -469,7 +472,7 @@ jobs:
         run: |
           docker exec -t ${{ env.container_name }} /bin/bash -ce '
           set -x
-          OPS_DST=paddlefleet_ops-0.0.0-cp310-cp310-linux_x86_64.whl
+          OPS_DST=paddlefleet_ops-0.0.0-cp312-cp312-linux_x86_64.whl
           wget -q --no-proxy "https://paddle-whl.bj.bcebos.com/nightly/cu129/paddlefleet-ops/${OPS_DST}" -O dist/${OPS_DST}
           '
 
@@ -506,6 +509,7 @@ jobs:
       PIP_CACHE_DIR: /root/.cache/pip
       CACHE_DIR: /root/.cache
       TASK: paddle-fleet-CI-${{ github.event.pull_request.number }}-single-card-test
+      PY_VERSION: "3.12"
     steps:
       - name: Determine the runner
         run: |
@@ -515,7 +519,7 @@ jobs:
       - name: Check docker image and run container
         env:
           GPU_DEVICES: ${{ env.GPU_DEVICES }}
-          docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-dev"
+          docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-py312-dev"
         run: |
           container_name=${TASK}-$(date +%Y%m%d-%H%M%S)
           echo "container_name=${container_name}" >> ${{ github.env }}
@@ -535,6 +539,7 @@ jobs:
             -e no_proxy \
             -e CI_name \
             -e PIP_CACHE_DIR \
+            -e PY_VERSION \
             -e work_dir \
             -e GITHUB_SHA="${{ github.event.pull_request.head.sha }}" \
             -e GITHUB_HEAD_REF="${{ github.head_ref }}" \
@@ -570,12 +575,12 @@ jobs:
           mkdir -p /PaddlePaddle
           cd /PaddlePaddle
           echo "Downloading Paddle.tar.gz from cfs"
-          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --no-check-certificate
+          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --no-check-certificate
           export UV_SKIP_WHEEL_FILENAME_CHECK=1 #This environment variable allows installing the latest commit-level whl package of Paddle.
           export UV_NO_SYNC=1 # This environment variable prevents uv sync from being executed when running un run.
           export UV_HTTP_TIMEOUT=300
           pip uninstall paddlepaddle-gpu -y
-          pip install paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
+          pip install paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
           echo "paddlefleet commit:"
           python -c "import paddlefleet; print(paddlefleet.version.commit)"
           '
@@ -624,7 +629,8 @@ jobs:
     env:
       PIP_CACHE_DIR: /root/.cache/pip
       TASK: paddle-fleet-CI-${{ github.event.pull_request.number }}-multi-card_test
-      docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-dev"
+      docker_image: "ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:ubuntu24-cuda129-py312-dev"
+      PY_VERSION: "3.12"
     steps:
       - name: Check docker image and run container
         run: |
@@ -647,6 +653,7 @@ jobs:
             -e no_proxy \
             -e CI_name \
             -e PIP_CACHE_DIR \
+            -e PY_VERSION \
             -e work_dir \
             -e GITHUB_SHA="${{ github.event.pull_request.head.sha }}" \
             -e GITHUB_HEAD_REF="${{ github.head_ref }}" \
@@ -683,12 +690,12 @@ jobs:
           mkdir -p /PaddlePaddle
           cd /PaddlePaddle
           echo "Downloading Paddle.tar.gz from cfs"
-          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --no-check-certificate
+          wget -q --tries=5 --no-proxy https://paddle-github-action.bj.bcebos.com/PR/h-coverage/${PR_ID}/${COMMIT_ID}/paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --no-check-certificate
           export UV_SKIP_WHEEL_FILENAME_CHECK=1 #This environment variable allows installing the latest commit-level whl package of Paddle.
           export UV_NO_SYNC=1 # This environment variable prevents uv sync from being executed when running un run.
           export UV_HTTP_TIMEOUT=300
           pip uninstall paddlepaddle-gpu -y
-          pip install paddlepaddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
+          pip install paddlepaddle_gpu-0.0.0-cp312-cp312-linux_x86_64.whl --force-reinstall --extra-index-url=https://www.paddlepaddle.org.cn/packages/nightly/cu129/
           echo "paddlefleet commit:"
           python -c "import paddlefleet; print(paddlefleet.version.commit)"
           '
diff --git a/test/test_flashmask_ci/flashmask_gt.json b/test/test_flashmask_ci/flashmask_gt.json
index e644d6e4e2faff..768a3c2f1df60c 100644
--- a/test/test_flashmask_ci/flashmask_gt.json
+++ b/test/test_flashmask_ci/flashmask_gt.json
@@ -19,16 +19,16 @@
   "gen_startend_row_indices7-1-8192-33792-2-1-1-80-80-3-dtype0": "6f8a11a2c3cbf189a274d9879e4f7e74",
   "gen_startend_row_indices8-1-8192-33792-2-1-1-80-80-3-dtype0": "223aec02c8e609f149b66eee8ef85f16",
   "gen_startend_row_indices9-1-8192-33792-2-1-1-80-80-3-dtype0": "276e98f28141a669d59e4656e02b3c7a",
-  "gen_startend_row_indices0-1-8192-33792-2-1-1-64-64-3-dtype0": "832fe895d299ba62bb3cf2a1c585090b",
-  "gen_startend_row_indices1-1-8192-33792-2-1-1-64-64-3-dtype0": "4dd468c9e48c0c9fb2444856c9d01b44",
-  "gen_startend_row_indices2-1-8192-33792-2-1-1-64-64-3-dtype0": "c7fbf1816774c1553fc506fa805baced",
-  "gen_startend_row_indices3-1-8192-33792-2-1-1-64-64-3-dtype0": "4dd468c9e48c0c9fb2444856c9d01b44",
-  "gen_startend_row_indices4-1-8192-33792-2-1-1-64-64-3-dtype0": "c67dcd5e0d4523268e1cb75220826fa5",
-  "gen_startend_row_indices5-1-8192-33792-2-1-1-64-64-3-dtype0": "4dd468c9e48c0c9fb2444856c9d01b44",
-  "gen_startend_row_indices6-1-8192-33792-2-1-1-64-64-3-dtype0": "e147c70967201739ae835f07c9d65bdd",
-  "gen_startend_row_indices7-1-8192-33792-2-1-1-64-64-3-dtype0": "e147c70967201739ae835f07c9d65bdd",
-  "gen_startend_row_indices8-1-8192-33792-2-1-1-64-64-3-dtype0": "e3885302eb4fbf8f930c018d1ab024f4",
-  "gen_startend_row_indices9-1-8192-33792-2-1-1-64-64-3-dtype0": "4dd468c9e48c0c9fb2444856c9d01b44",
+  "gen_startend_row_indices0-1-8192-33792-2-1-1-64-64-3-dtype0": "de0e0438bf1929e4dc145a9cae6c65cb",
+  "gen_startend_row_indices1-1-8192-33792-2-1-1-64-64-3-dtype0": "6354d399ab9a4a13dd6e35be6ca608cc",
+  "gen_startend_row_indices2-1-8192-33792-2-1-1-64-64-3-dtype0": "8d54745393a55eb2eb697c309b9c7985",
+  "gen_startend_row_indices3-1-8192-33792-2-1-1-64-64-3-dtype0": "6354d399ab9a4a13dd6e35be6ca608cc",
+  "gen_startend_row_indices4-1-8192-33792-2-1-1-64-64-3-dtype0": "136464684753a961677d6ac95cb12363",
+  "gen_startend_row_indices5-1-8192-33792-2-1-1-64-64-3-dtype0": "6354d399ab9a4a13dd6e35be6ca608cc",
+  "gen_startend_row_indices6-1-8192-33792-2-1-1-64-64-3-dtype0": "3c758670d54ee7729af549a6bfbef101",
+  "gen_startend_row_indices7-1-8192-33792-2-1-1-64-64-3-dtype0": "3c758670d54ee7729af549a6bfbef101",
+  "gen_startend_row_indices8-1-8192-33792-2-1-1-64-64-3-dtype0": "3ac600b75cf803abd03cc2124082ec62",
+  "gen_startend_row_indices9-1-8192-33792-2-1-1-64-64-3-dtype0": "6354d399ab9a4a13dd6e35be6ca608cc",
   "gen_startend_row_indices0-2840-32-32-16-4-1-128-128-3-dtype0": "75d4604f539bdeef39496544e3955008",
   "gen_startend_row_indices1-2840-32-32-16-4-1-128-128-3-dtype0": "1daf514cc698acb8449338629b4991e2",
   "gen_startend_row_indices2-2840-32-32-16-4-1-128-128-3-dtype0": "8624e86be21843c69bebde6e59c616ee",
@@ -109,16 +109,16 @@
   "gen_startend_row_indices7-1-300-300-16-16-1-80-80-3-dtype0": "1c2dfc027a7d1ff8a157f3158417f4a6",
   "gen_startend_row_indices8-1-300-300-16-16-1-80-80-3-dtype0": "83e8b2752bee81fac3ec6d51761c1459",
   "gen_startend_row_indices9-1-300-300-16-16-1-80-80-3-dtype0": "3d6991c67167de52846d75bafd637eb6",
-  "gen_startend_row_indices0-1-300-300-16-16-1-64-64-3-dtype0": "c3d5431321250abeaddc1237172b9e5c",
-  "gen_startend_row_indices1-1-300-300-16-16-1-64-64-3-dtype0": "6ff85ceae3f4b74b73fcec2d4b05eaa7",
-  "gen_startend_row_indices2-1-300-300-16-16-1-64-64-3-dtype0": "c16b31de4b90f5c01269628ceca82235",
-  "gen_startend_row_indices3-1-300-300-16-16-1-64-64-3-dtype0": "f74686d25f3e034b4166853324a3b237",
-  "gen_startend_row_indices4-1-300-300-16-16-1-64-64-3-dtype0": "c4834536e28f3fda10fc38a269023e76",
-  "gen_startend_row_indices5-1-300-300-16-16-1-64-64-3-dtype0": "7b7fa4996eb2a037180601de80da8fb2",
-  "gen_startend_row_indices6-1-300-300-16-16-1-64-64-3-dtype0": "6f3f0cca7f0242e49b9adb8fab16a55c",
-  "gen_startend_row_indices7-1-300-300-16-16-1-64-64-3-dtype0": "ee85640576bbc7089a697ef68d452263",
-  "gen_startend_row_indices8-1-300-300-16-16-1-64-64-3-dtype0": "e7fd028a89db9e925d9fe06a37ae1749",
-  "gen_startend_row_indices9-1-300-300-16-16-1-64-64-3-dtype0": "9cb6fd1607399c3e4767616021ad1955",
+  "gen_startend_row_indices0-1-300-300-16-16-1-64-64-3-dtype0": "3c4a8e5526e4020791e09da80b9b443f",
+  "gen_startend_row_indices1-1-300-300-16-16-1-64-64-3-dtype0": "e3bc435c882efb28862836dd4a84912b",
+  "gen_startend_row_indices2-1-300-300-16-16-1-64-64-3-dtype0": "52dac3b8053b40d2f73c5df954c192b3",
+  "gen_startend_row_indices3-1-300-300-16-16-1-64-64-3-dtype0": "da310e73d0de2062f7a0ed870b663548",
+  "gen_startend_row_indices4-1-300-300-16-16-1-64-64-3-dtype0": "63354fb212f95a81fa8e87689d4d87e1",
+  "gen_startend_row_indices5-1-300-300-16-16-1-64-64-3-dtype0": "2d6d451c261b2f802824cb7594d1b9f8",
+  "gen_startend_row_indices6-1-300-300-16-16-1-64-64-3-dtype0": "414e667be64dc78448d1284f04ceeec3",
+  "gen_startend_row_indices7-1-300-300-16-16-1-64-64-3-dtype0": "90aceddc48e9273e27694d2f6162e659",
+  "gen_startend_row_indices8-1-300-300-16-16-1-64-64-3-dtype0": "0b4ee542fa7732a235e1efc350bba901",
+  "gen_startend_row_indices9-1-300-300-16-16-1-64-64-3-dtype0": "ac21da119782ac451fbb7e0ae6789cde",
   "gen_startend_row_indices0-1-300-300-16-16-16-128-128-3-dtype0": "e829012daa321ea9f1956bc264e4a4f6",
   "gen_startend_row_indices1-1-300-300-16-16-16-128-128-3-dtype0": "60338d7811fe9bf2682ce40b72af631b",
   "gen_startend_row_indices2-1-300-300-16-16-16-128-128-3-dtype0": "225f3424253cc5d61c20aaa72e3717ae",
@@ -139,16 +139,16 @@
   "gen_startend_row_indices7-1-300-300-16-16-16-80-80-3-dtype0": "1c2dfc027a7d1ff8a157f3158417f4a6",
   "gen_startend_row_indices8-1-300-300-16-16-16-80-80-3-dtype0": "83e8b2752bee81fac3ec6d51761c1459",
   "gen_startend_row_indices9-1-300-300-16-16-16-80-80-3-dtype0": "998daf470bd932da27db90f4bccba87f",
-  "gen_startend_row_indices0-1-300-300-16-16-16-64-64-3-dtype0": "c3d5431321250abeaddc1237172b9e5c",
-  "gen_startend_row_indices1-1-300-300-16-16-16-64-64-3-dtype0": "6ff85ceae3f4b74b73fcec2d4b05eaa7",
-  "gen_startend_row_indices2-1-300-300-16-16-16-64-64-3-dtype0": "c16b31de4b90f5c01269628ceca82235",
-  "gen_startend_row_indices3-1-300-300-16-16-16-64-64-3-dtype0": "f74686d25f3e034b4166853324a3b237",
-  "gen_startend_row_indices4-1-300-300-16-16-16-64-64-3-dtype0": "c4834536e28f3fda10fc38a269023e76",
-  "gen_startend_row_indices5-1-300-300-16-16-16-64-64-3-dtype0": "7b7fa4996eb2a037180601de80da8fb2",
-  "gen_startend_row_indices6-1-300-300-16-16-16-64-64-3-dtype0": "6f3f0cca7f0242e49b9adb8fab16a55c",
-  "gen_startend_row_indices7-1-300-300-16-16-16-64-64-3-dtype0": "ee85640576bbc7089a697ef68d452263",
-  "gen_startend_row_indices8-1-300-300-16-16-16-64-64-3-dtype0": "e7fd028a89db9e925d9fe06a37ae1749",
-  "gen_startend_row_indices9-1-300-300-16-16-16-64-64-3-dtype0": "709e70e253e6e0e6204f63fde72e0d5b",
+  "gen_startend_row_indices0-1-300-300-16-16-16-64-64-3-dtype0": "3c4a8e5526e4020791e09da80b9b443f",
+  "gen_startend_row_indices1-1-300-300-16-16-16-64-64-3-dtype0": "e3bc435c882efb28862836dd4a84912b",
+  "gen_startend_row_indices2-1-300-300-16-16-16-64-64-3-dtype0": "52dac3b8053b40d2f73c5df954c192b3",
+  "gen_startend_row_indices3-1-300-300-16-16-16-64-64-3-dtype0": "da310e73d0de2062f7a0ed870b663548",
+  "gen_startend_row_indices4-1-300-300-16-16-16-64-64-3-dtype0": "63354fb212f95a81fa8e87689d4d87e1",
+  "gen_startend_row_indices5-1-300-300-16-16-16-64-64-3-dtype0": "2d6d451c261b2f802824cb7594d1b9f8",
+  "gen_startend_row_indices6-1-300-300-16-16-16-64-64-3-dtype0": "414e667be64dc78448d1284f04ceeec3",
+  "gen_startend_row_indices7-1-300-300-16-16-16-64-64-3-dtype0": "90aceddc48e9273e27694d2f6162e659",
+  "gen_startend_row_indices8-1-300-300-16-16-16-64-64-3-dtype0": "0b4ee542fa7732a235e1efc350bba901",
+  "gen_startend_row_indices9-1-300-300-16-16-16-64-64-3-dtype0": "a3ea8a49219141eff99dd532b7907f4a",
   "gen_startend_row_indices0-1-128-127-1-1-1-128-128-3-dtype0": "da0eee3fdf461e6f015b5360595c30ef",
   "gen_startend_row_indices1-1-128-127-1-1-1-128-128-3-dtype0": "ec86715f0f0ba520fab44cf4e05e0f60",
   "gen_startend_row_indices2-1-128-127-1-1-1-128-128-3-dtype0": "ea4407cbfad4b8d86c2031430011d3f4",
@@ -169,16 +169,16 @@
   "gen_startend_row_indices7-1-128-127-1-1-1-80-80-3-dtype0": "30e47c3513d68703fe233a8f96006e96",
   "gen_startend_row_indices8-1-128-127-1-1-1-80-80-3-dtype0": "f8f8efef3816d5686c84297003e08677",
   "gen_startend_row_indices9-1-128-127-1-1-1-80-80-3-dtype0": "b318eb75c32d18dc452f4c5eaa303c9d",
-  "gen_startend_row_indices0-1-128-127-1-1-1-64-64-3-dtype0": "1ae7d925db2f3d4728c347ff5a98edd8",
-  "gen_startend_row_indices1-1-128-127-1-1-1-64-64-3-dtype0": "1cb070d2b56c09d1c2a7473616ca8fe4",
+  "gen_startend_row_indices0-1-128-127-1-1-1-64-64-3-dtype0": "4031fbc9e9d39f6ad46ea19baf306522",
+  "gen_startend_row_indices1-1-128-127-1-1-1-64-64-3-dtype0": "f1504910842d96eb11a9ab2c8281660a",
   "gen_startend_row_indices2-1-128-127-1-1-1-64-64-3-dtype0": "c59f6e18d0e3d8f55794cbcd4ca086fe",
-  "gen_startend_row_indices3-1-128-127-1-1-1-64-64-3-dtype0": "d14ea041ff8ccec0a09c25baaa081803",
+  "gen_startend_row_indices3-1-128-127-1-1-1-64-64-3-dtype0": "6d378a97f83a37dcd69d0ac4449c6988",
   "gen_startend_row_indices4-1-128-127-1-1-1-64-64-3-dtype0": "78e535b25cef325d6f47cb7b9ee112e1",
-  "gen_startend_row_indices5-1-128-127-1-1-1-64-64-3-dtype0": "85818416e2c045a03f404b652e8d45d0",
+  "gen_startend_row_indices5-1-128-127-1-1-1-64-64-3-dtype0": "c5dd3a733b3662ac5ce34fd8329babb0",
   "gen_startend_row_indices6-1-128-127-1-1-1-64-64-3-dtype0": "927e978de57a5595471f98bc557949e7",
   "gen_startend_row_indices7-1-128-127-1-1-1-64-64-3-dtype0": "00cb19241518fb50f21ed3c609026290",
-  "gen_startend_row_indices8-1-128-127-1-1-1-64-64-3-dtype0": "401b48780b94781e4712fbda88786c99",
-  "gen_startend_row_indices9-1-128-127-1-1-1-64-64-3-dtype0": "6cd6fdf42043bab1c4a0b328502550ab",
+  "gen_startend_row_indices8-1-128-127-1-1-1-64-64-3-dtype0": "2676428b36aca250f277d39ddbd51fd0",
+  "gen_startend_row_indices9-1-128-127-1-1-1-64-64-3-dtype0": "8555f39d0c338602ae527049f3bb9c80",
   "gen_startend_row_indices0-2-16384-16383-4-1-1-128-128-3-dtype0": "028adc4ce9cd9b28ee5f33e828f4c8f9",
   "gen_startend_row_indices1-2-16384-16383-4-1-1-128-128-3-dtype0": "d7b370be62693319b6435f22a0ba3590",
   "gen_startend_row_indices2-2-16384-16383-4-1-1-128-128-3-dtype0": "463d09dd28b2ebdda6dd1fcf0f0bdd28",
@@ -199,16 +199,16 @@
   "gen_startend_row_indices7-2-16384-16383-4-1-1-80-80-3-dtype0": "8ff4f1415529010c719a87766babf63a",
   "gen_startend_row_indices8-2-16384-16383-4-1-1-80-80-3-dtype0": "e7ef54354c7a4d8d2dd66dbb822d1897",
   "gen_startend_row_indices9-2-16384-16383-4-1-1-80-80-3-dtype0": "84972da6ab2f99c01bb637ffeb621696",
-  "gen_startend_row_indices0-2-16384-16383-4-1-1-64-64-3-dtype0": "d3865daa03276277b52bcf837c940472",
-  "gen_startend_row_indices1-2-16384-16383-4-1-1-64-64-3-dtype0": "1a3cc3a4c03aa69e40676a47d3966f81",
-  "gen_startend_row_indices2-2-16384-16383-4-1-1-64-64-3-dtype0": "54e656dedb77473f30c91b7fc5efb622",
-  "gen_startend_row_indices3-2-16384-16383-4-1-1-64-64-3-dtype0": "97f4aeed9f570b189b7e2e95b54d094b",
-  "gen_startend_row_indices4-2-16384-16383-4-1-1-64-64-3-dtype0": "06a623fba8de963ed78a5dbd667cba52",
-  "gen_startend_row_indices5-2-16384-16383-4-1-1-64-64-3-dtype0": "ea4e4de2f6aaa076998ac1cd89d7cbf3",
-  "gen_startend_row_indices6-2-16384-16383-4-1-1-64-64-3-dtype0": "e044ff6bb7c06867f1d23de52f72018b",
-  "gen_startend_row_indices7-2-16384-16383-4-1-1-64-64-3-dtype0": "a51955812865e8e98356f135d1bd3abd",
-  "gen_startend_row_indices8-2-16384-16383-4-1-1-64-64-3-dtype0": "562b6f426a76df618a1e845b0f2ab85f",
-  "gen_startend_row_indices9-2-16384-16383-4-1-1-64-64-3-dtype0": "989deffab5b0ccf074f90adc8a8fecbd",
+  "gen_startend_row_indices0-2-16384-16383-4-1-1-64-64-3-dtype0": "b1143dff78a476ed923e58c5d6844e81",
+  "gen_startend_row_indices1-2-16384-16383-4-1-1-64-64-3-dtype0": "f825f1299dba84f2ce4330de3b5e29d4",
+  "gen_startend_row_indices2-2-16384-16383-4-1-1-64-64-3-dtype0": "00527259e57a4ef8cecbad406be732e0",
+  "gen_startend_row_indices3-2-16384-16383-4-1-1-64-64-3-dtype0": "25746353628ff2264a06456fbcbfe235",
+  "gen_startend_row_indices4-2-16384-16383-4-1-1-64-64-3-dtype0": "860dec17ecc61e30d30a38cebdcef726",
+  "gen_startend_row_indices5-2-16384-16383-4-1-1-64-64-3-dtype0": "10291567208228555ef82b269d3a8623",
+  "gen_startend_row_indices6-2-16384-16383-4-1-1-64-64-3-dtype0": "5ca25f3763cda372cc5856ac89bea719",
+  "gen_startend_row_indices7-2-16384-16383-4-1-1-64-64-3-dtype0": "77210e4dbc776a272bd7ce9a235891cd",
+  "gen_startend_row_indices8-2-16384-16383-4-1-1-64-64-3-dtype0": "47d3c84b4ff075b053b7e8075185685e",
+  "gen_startend_row_indices9-2-16384-16383-4-1-1-64-64-3-dtype0": "ca64e369235b9a1a001c7e174421ffe9",
   "gt_commit_id": "61459524eb45ac1fd8ec1d75be38ad47c410c756",
   "gt_commit_msg": "add blockmask for FlashMaskV3 (#89)"
 }
diff --git a/test/test_flashmask_ci/run.sh b/test/test_flashmask_ci/run.sh
index b6bee0291b7b10..8a0f14eb1e353f 100644
--- a/test/test_flashmask_ci/run.sh
+++ b/test/test_flashmask_ci/run.sh
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+set -e
+
 python -m pytest test_flashmask_ci.py -v
 python -m pytest test_fwd_md5sum.py -v
 python -m pytest test_flashmask_group.py -v