Merge pull request #1836 from reguly/feature/intelGPU

gpu: Adding DPCPP compiler support

devito/arch/archinfo.py

@@ -15,11 +15,11 @@
 __all__ = ['platform_registry', 'get_cpu_info', 'get_gpu_info', 'get_nvidia_cc',
            'check_cuda_runtime',
            'Platform', 'Cpu64', 'Intel64', 'Amd', 'Arm', 'Power', 'Device',
-           'NvidiaDevice', 'AmdDevice',
+           'NvidiaDevice', 'AmdDevice', 'IntelDevice',
            'INTEL64', 'SNB', 'IVB', 'HSW', 'BDW', 'SKX', 'KNL', 'KNL7210',  # Intel
            'AMD', 'ARM', 'M1',  # ARM
            'POWER8', 'POWER9',  # Other loosely supported CPU architectures
-           'AMDGPUX', 'NVIDIAX']  # GPUs
+           'AMDGPUX', 'NVIDIAX', 'INTELGPUX']  # GPUs
 
 
 @memoized_func
@@ -600,6 +600,13 @@ def memavail(self, deviceid=0):
             return None
 
 
+class IntelDevice(Device):
+
+    @cached_property
+    def march(self):
+        return ''
+
+
 class NvidiaDevice(Device):
 
     @cached_property
@@ -665,6 +672,7 @@ def march(cls):
 # Devices
 NVIDIAX = NvidiaDevice('nvidiaX')
 AMDGPUX = AmdDevice('amdgpuX')
+INTELGPUX = IntelDevice('intelgpuX')
 
 
 platform_registry = {
@@ -685,7 +693,8 @@ def march(cls):
     'power8': POWER8,
     'power9': POWER9,
     'nvidiaX': NVIDIAX,  # Generic NVidia GPU
-    'amdgpuX': AMDGPUX   # Generic AMD GPU
+    'amdgpuX': AMDGPUX,   # Generic AMD GPU
+    'intelgpuX': INTELGPUX   # Generic Intel GPU
 }
 """
 Registry dict for deriving Platform classes according to the environment variable

devito/arch/compiler.py

@@ -457,6 +457,22 @@ def __lookup_cmds__(self):
         self.MPICXX = 'mpicxx'
 
 
+class DPCPPCompiler(Compiler):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        self.cflags += ['-qopenmp', '-fopenmp-targets=spir64']
+
+    def __lookup_cmds__(self):
+        # OneAPI Base Kit comes with dpcpp/icpx, both are clang++,
+        # and icx, which is clang
+        self.CC = 'icx'
+        self.CXX = 'icpx'
+        self.MPICC = 'mpic++'
+        self.MPICXX = 'mpicxx'
+
+
 class PGICompiler(Compiler):
 
     def __init__(self, *args, **kwargs):
@@ -641,6 +657,7 @@ def __lookup_cmds__(self):
     'icc': IntelCompiler,
     'intel-knl': IntelKNLCompiler,
     'knl': IntelKNLCompiler,
+    'dpcpp': DPCPPCompiler,
 }
 """
 Registry dict for deriving Compiler classes according to the environment variable

devito/passes/iet/languages/openmp.py

@@ -1,7 +1,7 @@
 import cgen as c
 from sympy import Not
 
-from devito.arch import AMDGPUX, NVIDIAX
+from devito.arch import AMDGPUX, NVIDIAX, INTELGPUX
 from devito.ir import (Call, Conditional, List, Prodder, ParallelIteration,
                        ParallelBlock, PointerCast, While, FindSymbols)
 from devito.passes.iet.definitions import DataManager, DeviceAwareDataManager
@@ -122,6 +122,7 @@ class OmpBB(PragmaLangBB):
         # Platform mapping
         AMDGPUX: None,
         NVIDIAX: None,
+        INTELGPUX: None,
         # Runtime library
         'init': None,
         'thread-num': lambda retobj=None: