Skip to content

Commit 9d9703c

Browse files
committed
compiler: add missing Copenmp registry entries
1 parent f45a2da commit 9d9703c

2 files changed

Lines changed: 11 additions & 5 deletions

File tree

devito/core/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,41 +33,49 @@
3333
# Register CPU Operators
3434
operator_registry.add(Cpu64CustomOperator, Cpu64, 'custom', 'C')
3535
operator_registry.add(Cpu64CustomOperator, Cpu64, 'custom', 'openmp')
36+
operator_registry.add(Cpu64CustomOperator, Cpu64, 'custom', 'Copenmp')
3637
operator_registry.add(Cpu64CustomCXXOperator, Cpu64, 'custom', 'CXX')
3738
operator_registry.add(Cpu64CustomCXXOperator, Cpu64, 'custom', 'CXXopenmp')
3839

3940
operator_registry.add(Cpu64NoopCOperator, Cpu64, 'noop', 'C')
4041
operator_registry.add(Cpu64NoopOmpOperator, Cpu64, 'noop', 'openmp')
42+
operator_registry.add(Cpu64NoopOmpOperator, Cpu64, 'noop', 'Copenmp')
4143
operator_registry.add(Cpu64CXXNoopCOperator, Cpu64, 'noop', 'CXX')
4244
operator_registry.add(Cpu64CXXNoopOmpOperator, Cpu64, 'noop', 'CXXopenmp')
4345

4446
operator_registry.add(Cpu64AdvCOperator, Cpu64, 'advanced', 'C')
4547
operator_registry.add(Cpu64AdvOmpOperator, Cpu64, 'advanced', 'openmp')
48+
operator_registry.add(Cpu64AdvOmpOperator, Cpu64, 'advanced', 'Copenmp')
4649
operator_registry.add(Cpu64AdvCXXOperator, Cpu64, 'advanced', 'CXX')
4750
operator_registry.add(Cpu64AdvCXXOmpOperator, Cpu64, 'advanced', 'CXXopenmp')
4851

4952
operator_registry.add(Cpu64FsgCOperator, Cpu64, 'advanced-fsg', 'C')
5053
operator_registry.add(Cpu64FsgOmpOperator, Cpu64, 'advanced-fsg', 'openmp')
54+
operator_registry.add(Cpu64FsgOmpOperator, Cpu64, 'advanced-fsg', 'Copenmp')
5155
operator_registry.add(Cpu64FsgCXXOperator, Cpu64, 'advanced-fsg', 'CXX')
5256
operator_registry.add(Cpu64FsgCXXOmpOperator, Cpu64, 'advanced-fsg', 'CXXopenmp')
5357

5458
operator_registry.add(Intel64AdvCOperator, Intel64, 'advanced', 'C')
5559
operator_registry.add(Intel64AdvOmpOperator, Intel64, 'advanced', 'openmp')
60+
operator_registry.add(Intel64AdvOmpOperator, Intel64, 'advanced', 'Copenmp')
5661
operator_registry.add(Intel64CXXAdvCOperator, Intel64, 'advanced', 'CXX')
5762
operator_registry.add(Intel64AdvCXXOmpOperator, Intel64, 'advanced', 'CXXopenmp')
5863

5964
operator_registry.add(Intel64FsgCOperator, Intel64, 'advanced-fsg', 'C')
6065
operator_registry.add(Intel64FsgOmpOperator, Intel64, 'advanced-fsg', 'openmp')
66+
operator_registry.add(Intel64FsgOmpOperator, Intel64, 'advanced-fsg', 'Copenmp')
6167
operator_registry.add(Intel64FsgCXXOperator, Intel64, 'advanced-fsg', 'CXX')
6268
operator_registry.add(Intel64FsgCXXOmpOperator, Intel64, 'advanced-fsg', 'CXXopenmp')
6369

6470
operator_registry.add(ArmAdvCOperator, Arm, 'advanced', 'C')
6571
operator_registry.add(ArmAdvOmpOperator, Arm, 'advanced', 'openmp')
72+
operator_registry.add(ArmAdvOmpOperator, Arm, 'advanced', 'Copenmp')
6673
operator_registry.add(ArmAdvCXXOperator, Arm, 'advanced', 'CXX')
6774
operator_registry.add(ArmAdvCXXOmpOperator, Arm, 'advanced', 'CXXopenmp')
6875

6976
operator_registry.add(PowerAdvCOperator, Power, 'advanced', 'C')
7077
operator_registry.add(PowerAdvOmpOperator, Power, 'advanced', 'openmp')
78+
operator_registry.add(PowerAdvOmpOperator, Power, 'advanced', 'Copenmp')
7179
operator_registry.add(PowerCXXAdvCOperator, Power, 'advanced', 'CXX')
7280
operator_registry.add(PowerAdvCXXOmpOperator, Power, 'advanced', 'CXXopenmp')
7381

devito/operator/operator.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1403,11 +1403,9 @@ def parse_kwargs(**kwargs):
14031403
kwargs['language'] = language
14041404
elif kwopenmp is not None:
14051405
# Handle deprecated `openmp` kwarg for backward compatibility
1406-
if configuration['language'] in ['C', 'CXX']:
1407-
lang = configuration['language']
1408-
kwargs['language'] = f'{lang}openmp' if openmp else lang
1409-
else:
1410-
kwargs['language'] = 'openmp' if openmp else 'C'
1406+
omp = {'C': 'openmp', 'CXX': 'CXXopenmp'}.get(configuration['language'],
1407+
'openmp')
1408+
kwargs['language'] = omp if openmp else 'C'
14111409
else:
14121410
kwargs['language'] = configuration['language']
14131411

0 commit comments

Comments
 (0)