Skip to content

[compiler] fix unfuse batch norm with fp16 type #351

[compiler] fix unfuse batch norm with fp16 type

[compiler] fix unfuse batch norm with fp16 type #351

name: TorchFrontend CI
on:
pull_request:
branches:
- main
paths:
- ".github/workflows/torch-frontend-ci.yaml"
- "frontends/torch-frontend/**"
- "compiler/dialects/include/byteir/Dialect/Ccl/**"
- "compiler/dialects/lib/Dialect/Ccl/**"
- "!**/**.md"
- "!frontends/torch-frontend/torch-frontend/python/version.txt"
workflow_dispatch:
# Ensure that only a single job or workflow using the same
# concurrency group will run at a time. This would cancel
# any in-progress jobs in the same github workflow and github
# ref (e.g. refs/heads/main or refs/pull/<pr_number>/merge).
concurrency:
# A PR number if a pull request and otherwise the commit hash. This cancels
# queued and in-progress runs for the same PR (presubmit) or commit
# (postsubmit). The workflow name is prepended to avoid conflicts between
# different workflows.
group: ${{ github.workflow }}-${{ github.event.number || github.sha }}
cancel-in-progress: true
jobs:
torch_frontend_build_and_test:
name: torch-frontend CI
runs-on: self-hosted
container:
image: localhost:5000/byteir-ci
volumes:
- /data00/llvm_libraries:/data00/llvm_libraries
- /data00/mhlo_libraries:/data00/mhlo_libraries
options: --runtime=nvidia --gpus all
steps:
- name: clear workspace
run: rm -rf $GITHUB_WORKSPACE && mkdir $GITHUB_WORKSPACE
- name: Checkout byteir repo
uses: actions/checkout@v3
- name: Build and test TorchFrontend
run: python3 -m pip install -r ./frontends/torch-frontend/torch-cuda-requirements.txt && ./frontends/torch-frontend/scripts/build_and_test.sh
shell: bash