Skip to content

[bugfix] narrow the fx node name to str in dense graph rewrite #798

[bugfix] narrow the fx node name to str in dense graph rewrite

[bugfix] narrow the fx node name to str in dense graph rewrite #798

Workflow file for this run

name: Unit Test H20 CI
on:
pull_request:
types: [opened, reopened, synchronize]
workflow_dispatch:
concurrency:
group: unittest-h20-ci-${{ github.event.pull_request.number }}
cancel-in-progress: true
jobs:
ci-test:
# `container:` would inject -v /var/run/docker.sock which DSW authZ blocks here.
runs-on: tzrec-h20-runner
steps:
- name: FetchCommit ${{ github.event.pull_request.head.sha }}
uses: actions/checkout@v4
with:
ref: ${{ github.event.pull_request.head.sha }}
path: run_${{ github.run_id }}
- name: RunUnitTestH20CI
id: run_unittest_h20_ci
run: |
WORK_ROOT="$(dirname "$RUNNER_WORKSPACE")"
# Resolve symlink: runner self-update makes externals a symlink, which DSW authZ rejects as a mount source.
EXTERNALS_ROOT="$(readlink -f "$(dirname "$WORK_ROOT")/externals")"
docker run --rm --name "h20-ci-${{ github.run_id }}" \
--workdir /__w/TorchEasyRec/TorchEasyRec \
--gpus all --shm-size=512g --ulimit memlock=-1 \
-e HOME=/github/home \
-e GITHUB_ACTIONS=true \
-e CI=true \
-e CUDA_HOME=/usr/local/cuda-13 \
-e TRITON_CACHE_DIR=/__w/TorchEasyRec/tzrec-ci-cache/triton \
-e TORCHINDUCTOR_CACHE_DIR=/__w/TorchEasyRec/tzrec-ci-cache/inductor \
-v "$WORK_ROOT":/__w \
-v "$EXTERNALS_ROOT":/__e:ro \
-v "$WORK_ROOT/_temp":/__w/_temp \
-v "$WORK_ROOT/_actions":/__w/_actions \
-v "$WORK_ROOT/_tool":/__w/_tool \
-v "$WORK_ROOT/_temp/_github_home":/github/home \
-v "$WORK_ROOT/_temp/_github_workflow":/github/workflow \
mybigpai-public-registry.cn-beijing.cr.aliyuncs.com/easyrec/tzrec-devel:1.4 \
bash -c 'mkdir -p /__w/TorchEasyRec/tzrec-ci-cache/triton /__w/TorchEasyRec/tzrec-ci-cache/inductor && cd run_${{ github.run_id }} && bash scripts/ci/ci_test.sh --scope h20'
- name: StopDockerContainer
if: always()
run: |
docker stop --timeout=30 "h20-ci-${{ github.run_id }}" 2>/dev/null || true
docker rm -f "h20-ci-${{ github.run_id }}" 2>/dev/null || true