Skip to content

Commit

Permalink
Scheduler test: account for new trafo in SCCVectorPipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
MichaelSt98 committed Sep 10, 2024
1 parent b97c5f9 commit 502558f
Showing 1 changed file with 14 additions and 13 deletions.
27 changes: 14 additions & 13 deletions loki/batch/tests/test_scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -2889,21 +2889,22 @@ def test_pipeline_config_compose(config):
assert isinstance(pipeline, Pipeline)

# Check that the pipeline is correctly composed
assert len(pipeline.transformations) == 7
assert len(pipeline.transformations) == 8
assert type(pipeline.transformations[0]).__name__ == 'RemoveCodeTransformation'
assert type(pipeline.transformations[1]).__name__ == 'SCCBaseTransformation'
assert type(pipeline.transformations[2]).__name__ == 'SCCDevectorTransformation'
assert type(pipeline.transformations[3]).__name__ == 'SCCDemoteTransformation'
assert type(pipeline.transformations[4]).__name__ == 'SCCRevectorTransformation'
assert type(pipeline.transformations[5]).__name__ == 'SCCAnnotateTransformation'
assert type(pipeline.transformations[6]).__name__ == 'ModuleWrapTransformation'
assert type(pipeline.transformations[1]).__name__ == 'SCCFuseVerticalLoops'
assert type(pipeline.transformations[2]).__name__ == 'SCCBaseTransformation'
assert type(pipeline.transformations[3]).__name__ == 'SCCDevectorTransformation'
assert type(pipeline.transformations[4]).__name__ == 'SCCDemoteTransformation'
assert type(pipeline.transformations[5]).__name__ == 'SCCRevectorTransformation'
assert type(pipeline.transformations[6]).__name__ == 'SCCAnnotateTransformation'
assert type(pipeline.transformations[7]).__name__ == 'ModuleWrapTransformation'

# Check for some specified and default constructor flags
assert pipeline.transformations[0].call_names == ('dr_hook',)
assert pipeline.transformations[0].remove_imports is False
assert isinstance(pipeline.transformations[1].horizontal, Dimension)
assert pipeline.transformations[1].horizontal.size == 'KLON'
assert pipeline.transformations[1].horizontal.index == 'JL'
assert pipeline.transformations[1].directive == 'openacc'
assert pipeline.transformations[2].trim_vector_sections is True
assert pipeline.transformations[6].replace_ignore_items is True
assert isinstance(pipeline.transformations[2].horizontal, Dimension)
assert pipeline.transformations[2].horizontal.size == 'KLON'
assert pipeline.transformations[2].horizontal.index == 'JL'
assert pipeline.transformations[2].directive == 'openacc'
assert pipeline.transformations[3].trim_vector_sections is True
assert pipeline.transformations[7].replace_ignore_items is True

0 comments on commit 502558f

Please sign in to comment.