compiler: minor and flake8

xdslproject · Jun 19, 2024 · 9b3f6a8 · 9b3f6a8
1 parent d744348
commit 9b3f6a8
Show file tree

Hide file tree

Showing 5 changed files with 108 additions and 131 deletions.
diff --git a/devito/core/__init__.py b/devito/core/__init__.py
@@ -14,6 +14,10 @@
                              DeviceCustomOmpOperator, DeviceCustomAccOperator)
 from devito.operator.registry import operator_registry
 
+# Import XDSL Operators
+from devito.xdsl_core.xdsl_cpu import XdslnoopOperator, XdslAdvOperator
+from devito.xdsl_core.xdsl_gpu import XdslAdvDeviceOperator
+
 # Register CPU Operators
 operator_registry.add(Cpu64CustomOperator, Cpu64, 'custom', 'C')
 operator_registry.add(Cpu64CustomOperator, Cpu64, 'custom', 'openmp')
@@ -54,3 +58,11 @@
 operator_registry.add(DeviceFsgOmpOperator, Device, 'advanced-fsg', 'C')
 operator_registry.add(DeviceFsgOmpOperator, Device, 'advanced-fsg', 'openmp')
 operator_registry.add(DeviceFsgAccOperator, Device, 'advanced-fsg', 'openacc')
+
+# Register XDSL Operators
+operator_registry.add(XdslnoopOperator, Cpu64, 'xdsl-noop', 'C')
+operator_registry.add(XdslAdvOperator, Cpu64, 'xdsl-noop', 'openmp')
+
+operator_registry.add(XdslAdvOperator, Cpu64, 'xdsl', 'C')
+operator_registry.add(XdslAdvOperator, Cpu64, 'xdsl', 'openmp')
+operator_registry.add(XdslAdvDeviceOperator, Device, 'xdsl', 'openacc')