Skip to content

Commit

Permalink
Scheduler: Adjust pipeline_config_compose test to new Transformation
Browse files Browse the repository at this point in the history
  • Loading branch information
mlange05 committed Apr 12, 2024
1 parent df042ed commit bf9cb98
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions tests/test_scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -2870,11 +2870,11 @@ def test_pipeline_config_compose(config):
},
},
'preprocess': {
'classname': 'RemoveCallsTransformation',
'module': 'transformations.utility_routines',
'classname': 'RemoveCodeTransformation',
'module': 'loki.transform.transform_remove_code',
'options': {
'routines': 'dr_hook',
'include_intrinsics': True
'call_names': 'dr_hook',
'remove_imports': False
}
},
'postprocess': {
Expand Down Expand Up @@ -2905,7 +2905,7 @@ def test_pipeline_config_compose(config):

# Check that the pipeline is correctly composed
assert len(pipeline.transformations) == 7
assert type(pipeline.transformations[0]).__name__ == 'RemoveCallsTransformation'
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'
Expand All @@ -2914,7 +2914,8 @@ def test_pipeline_config_compose(config):
assert type(pipeline.transformations[6]).__name__ == 'ModuleWrapTransformation'

# Check for some specified and default constructor flags
assert pipeline.transformations[0].include_intrinsics is True
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'
Expand Down

0 comments on commit bf9cb98

Please sign in to comment.