Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Run test reshape separately
Browse files Browse the repository at this point in the history
TizianoDeMatteis committed Jun 10, 2021
1 parent cb0da61 commit 8acd5da
Showing 3 changed files with 10 additions and 2 deletions.
8 changes: 7 additions & 1 deletion .github/workflows/cpu-ci.yml
Original file line number Diff line number Diff line change
@@ -54,7 +54,13 @@ jobs:
- name: Test with pytest
env:
ORT_RELEASE: ${{ github.workspace }}/onnxruntime-daceml-patched
PYTEST_ARGS: --cov=daceml --cov-report=term --cov-report xml --cov-config=.coveragerc -m "not slow and not fpga and not gpu"
PYTEST_ARGS: --cov=daceml --cov-report=term --cov-report xml --cov-config=.coveragerc -m "not slow and not fpga and not gpu and not seq"
run: make test

- name: Seq Test with pytest
env:
ORT_RELEASE: ${{ github.workspace }}/onnxruntime-daceml-patched
PYTEST_ARGS: --cov=daceml --cov-report=term --cov-report xml --cov-config=.coveragerc -m "not slow and not fpga and not gpu and seq"
run: make test

- name: Test with doctest
1 change: 1 addition & 0 deletions pytest.ini
Original file line number Diff line number Diff line change
@@ -5,3 +5,4 @@ markers =
ort: marks tests that test onnxruntime ops (and sets the default implementation before executing that test)
gpu: marks tests that should only run when --gpu or --gpu-only are passed
fpga: marks tests for FPGA (deselect with '-m "not fpga"')
seq: mark tests that should run in a separate action
3 changes: 2 additions & 1 deletion tests/transformation/test_reshape_elimination.py
Original file line number Diff line number Diff line change
@@ -20,12 +20,13 @@ def forward(self, x):
return F.relu(x)


@pytest.mark.seq
@pytest.mark.pure
def test_reshape_elimination(sdfg_name):

ptmodel = Model()
x = torch.rand((100, 6, 12, 12))
dace_model = DaceModule(ptmodel, auto_optimize=False, sdfg_name=sdfg_name, cuda=False)
dace_model = DaceModule(ptmodel, auto_optimize=False, sdfg_name=sdfg_name)

def ApplyReshapeElimination(dace_module):
sdfg = dace_module.sdfg

0 comments on commit 8acd5da

Please sign in to comment.