Merge pull request #117 from PokhodenkoSA/numba-dppy

Add subtree numba-dppy

Author: PokhodenkoSA

numba-dppy/LICENSE

@@ -0,0 +1,23 @@
+Copyright (c) 2019-2020, Intel Corporation
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

numba-dppy/README.md

@@ -0,0 +1,21 @@
+# numba-dppy
+
+## Numba + dpCtl + dpNP = numba-dppy
+
+`numba-dppy` extends Numba with new backend for support compilation
+for Intel CPU and GPU architectures.
+
+For more information about Numba, see the Numba homepage:
+http://numba.pydata.org
+
+For more information about dpCtl, see the dpCtl homepage:
+https://intelpython.github.io/dpctl/
+
+For more information about dpNP, see the dpNP homepage:
+https://intelpython.github.io/dpnp/
+
+## Dependencies
+
+* numba
+* dpCtl
+* dpNP (optional)

numba-dppy/numba_dppy/.gitignore

@@ -0,0 +1,3 @@
+*.bc
+*.ll
+*.spirv

numba-dppy/numba_dppy/CHANGE_LOG

@@ -0,0 +1,7 @@
+NUMBA Version 0.48.0 + DPPL Version 0.3.0 (June 29, 2020)
+--------------------------------------------------------
+
+This release includes:
+
+* Caching of dppl.kernels which will improve performance.
+* Addition of support for Intel Advisor which will help in profiling applications.

numba-dppy/numba_dppy/__init__.py

@@ -0,0 +1,74 @@
+from llvmlite import binding as ll
+from llvmlite.llvmpy import core as lc
+
+from numba.core.codegen import BaseCPUCodegen, CodeLibrary
+from numba.core import utils
+
+
+SPIR_TRIPLE = {32: ' spir-unknown-unknown',
+               64: 'spir64-unknown-unknown'}
+
+SPIR_DATA_LAYOUT = {
+    32 : ('e-p:32:32-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:'
+          '256-v512:512-v1024:1024'),
+    64 : ('e-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-'
+          'v512:512-v1024:1024')
+}
+
+
+class SPIRVCodeLibrary(CodeLibrary):
+    def _optimize_functions(self, ll_module):
+        pass
+
+    def _optimize_final_module(self):
+        # Run some lightweight optimization to simplify the module.
+        pmb = ll.PassManagerBuilder()
+        pmb.opt_level = 1
+        pmb.disable_unit_at_a_time = False
+        pmb.disable_unroll_loops = True
+        pmb.loop_vectorize = False
+        pmb.slp_vectorize = False
+
+        pm = ll.ModulePassManager()
+        pmb.populate(pm)
+        pm.run(self._final_module)
+
+    def _finalize_specific(self):
+        # Fix global naming
+        for gv in self._final_module.global_variables:
+            if '.' in gv.name:
+                gv.name = gv.name.replace('.', '_')
+
+    def get_asm_str(self):
+        # Return nothing: we can only dump assembler code when it is later
+        # generated (in numba.dppl.compiler).
+        return None
+
+
+class JITSPIRVCodegen(BaseCPUCodegen):
+    """
+    This codegen implementation generates optimized SPIR 2.0
+    """
+
+    _library_class = SPIRVCodeLibrary
+
+    def _init(self, llvm_module):
+        assert list(llvm_module.global_variables) == [], "Module isn't empty"
+        self._data_layout = SPIR_DATA_LAYOUT[utils.MACHINE_BITS]
+        self._target_data = ll.create_target_data(self._data_layout)
+
+    def _create_empty_module(self, name):
+        ir_module = lc.Module(name)
+        ir_module.triple = SPIR_TRIPLE[utils.MACHINE_BITS]
+        if self._data_layout:
+            ir_module.data_layout = self._data_layout
+        return ir_module
+
+    def _module_pass_manager(self):
+        raise NotImplementedError
+
+    def _function_pass_manager(self, llvm_module):
+        raise NotImplementedError
+
+    def _add_module(self, module):
+        pass