Skip to content

[Operator] premute for reduction tensors & welford var_mean #47

[Operator] premute for reduction tensors & welford var_mean

[Operator] premute for reduction tensors & welford var_mean #47

Workflow file for this run

# This workflow will install Python dependencies, run tests and lint with a single version of Python
# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python
name: flag-gems-test
on:
push:
branches: [ "master" ]
pull_request:
branches: [ "master" ]
jobs:
container-unit-test:
runs-on: [self-hosted, docker]
timeout-minutes: 30
container:
image: localhost:5000/flag-gems-ci:v1.0
ports:
- 81
options: --gpus all --hostname flag-gems_cicd_ut
steps:
- name: checkout-code
uses: actions/checkout@v4
- name: unit_test-flag-gems
run: |
CUDA_VISIBLE_DEVICES=0 pytest -s tests/test_unary_pointwise_ops.py &
CUDA_VISIBLE_DEVICES=1 pytest -s tests/test_binary_pointwise_ops.py &
CUDA_VISIBLE_DEVICES=2 pytest -s tests/test_blas_ops.py &
CUDA_VISIBLE_DEVICES=3 pytest -s tests/test_reduction_ops.py &
CUDA_VISIBLE_DEVICES=4 pytest -s tests/test_special_ops.py && wait
container-model-test:
runs-on: [self-hosted, docker]
timeout-minutes: 5
container:
image: localhost:5000/flag-gems-ci:v1.0
ports:
- 82
options: --gpus all --hostname flag-gems_cicd_model -v /home/flaggems_cicd/huggingface_cache_bert:/__w/_temp/_github_home/.cache/huggingface
steps:
- name: checkout-code
uses: actions/checkout@v4
- name: examples-flag-gems
run: |
CUDA_VISIBLE_DEVICES=5 pytest -s examples/model_bert_test.py