Update Numba to 0.57

.github/workflows/rbc_test.yml

@@ -108,11 +108,19 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        python-version: ['3.10', '3.9', '3.8']
-        numba-version: ['0.56', '0.55']
+        python-version: ['3.10', '3.9']
+        numba-version: ['0.56']
         heavydb-version: ['6.2', '6.1', '6.0']
+        llvm-version: ['11.1']
         heavydb-from: [conda]
-        # include:
+        include:
+          - os: ubuntu-latest
+            python-version: '3.11'
+            numba-version: '0.57'
+            llvm-version: '14'
+            heavydb-version: '7.0'
+            docker-image: heavyai/heavyai-ee-cpu:7.0.0
+            heavydb-from: docker
         #   - os: ubuntu-latest
         #     python-version: '3.10'
         #     numba-version: '0.56'
@@ -140,8 +148,8 @@ jobs:
       - uses: conda-incubator/setup-miniconda@v2
         with:
           python-version: ${{ matrix.python-version }}
-          channel-priority: strict
-          channels: conda-forge
+          channel-priority: flexible
+          channels: conda-forge,numba
 
       - name: Install mamba
         shell: bash -l {0}
@@ -197,6 +205,9 @@ jobs:
           cat environment.yml > rbc_test.yaml
           echo "  - numba=${{ matrix.numba-version }}" >> rbc_test.yaml
           echo "  - python=${{ matrix.python-version }}" >> rbc_test.yaml
+          echo "  - clang=${{ matrix.llvm-version }}" >> rbc_test.yaml
+          echo "  - clangxx=${{ matrix.llvm-version }}" >> rbc_test.yaml
+
           mamba env create --file=rbc_test.yaml -n rbc
 
       - name: rbc conda config
@@ -229,9 +240,9 @@ jobs:
           pkill -f heavydb
           mamba run -n rbc pytest -sv -r A rbc/tests/ -x --ignore rbc/tests/heavydb
 
-      - name: Run rbc tests
+      - name: Run rbc tests [conda]
         shell: bash -l {0}
-        if: ${{ ! ( matrix.os == 'ubuntu-latest' && matrix.heavydb-from == 'docker' ) }}
+        if: matrix.heavydb-from == 'conda'
         env:
           HEAVYDB_SOURCE: ${{ matrix.heavydb-from }}
           EXPECTED_PYTHON_VERSION: ${{ matrix.python-version }}
@@ -290,11 +301,11 @@ jobs:
         - name: Set up Python
           uses: actions/setup-python@v2
           with:
-            python-version: 3.8
+            python-version: '3.10'
 
         - uses: conda-incubator/setup-miniconda@v2
           with:
-            python-version: 3.8
+            python-version: '3.10'
             channel-priority: strict
             channels: conda-forge
             environment-file: environment.yml

environment.yml

@@ -12,10 +12,10 @@ channels:
 dependencies:
   - python>=3.7
   - setuptools
-  - numba>=0.55
+  - numba>=0.56
   - llvmlite
-  - clang==11.1
-  - clangxx==11.1
+  - clang
+  - clangxx
   - pytest
   - packaging
   - tblib

rbc/errors.py

@@ -3,9 +3,10 @@
 """
 
 
-from rbc.utils import get_version
 from numba.core.errors import TypingError
 
+from rbc.utils import get_version
+
 
 class HeavyDBServerError(Exception):
     """

rbc/heavydb/array.py

@@ -46,12 +46,11 @@ def deepcopy(self, context, builder, val, retptr):
         ptr_type = self.dtype.members[0]
         element_size = int64_t(ptr_type.dtype.bitwidth // 8)
 
-        struct_load = builder.load(val, name='struct_load')
-        src = builder.extract_value(struct_load, 0, name='array_buff_ptr')
-        element_count = builder.extract_value(struct_load, 1, name='array_size')
-        is_null = builder.extract_value(struct_load, 2, name='array_is_null')
-
         zero, one, two = int32_t(0), int32_t(1), int32_t(2)
+        src = builder.load(builder.gep(val, [zero, zero]), name='array_buff_ptr')
+        element_count = builder.load(builder.gep(val, [zero, one]), name='array_size')
+        is_null = builder.load(builder.gep(val, [zero, two]), name='array_is_null')
+
         with builder.if_else(cgutils.is_true(builder, is_null)) as (then, otherwise):
             with then:
                 nullptr = cgutils.get_null_value(src.type)

rbc/irtools.py

@@ -152,6 +152,22 @@ def _get_host_cpu_features(self):
         # See https://github.com/xnd-project/rbc/issues/45
         remove_features = {
             (12, 12): [], (11, 11): [], (10, 10): [], (9, 9): [], (8, 8): [],
+            (14, 14): ['avx512pf', 'tsxldtrk', 'cx16', 'sahf', 'tbm',
+                       'avx512ifma', 'sha', 'crc32', 'fma4', 'vpclmulqdq',
+                       'prfchw', 'bmi2', 'cldemote', 'fsgsbase', 'ptwrite',
+                       'amxtile', 'uintr', 'gfni', 'popcnt', 'widekl', 'aes',
+                       'avx512bitalg', 'movdiri', 'xsaves', 'avx512er', 'avxvnni',
+                       'avx512fp16', 'avx512vnni', 'amxbf16', 'avx512vpopcntdq',
+                       'pconfig', 'clwb', 'avx512f', 'xsavec', 'clzero', 'pku',
+                       'mmx', 'lwp', 'rdpid', 'xop', 'rdseed', 'waitpkg', 'kl',
+                       'movdir64b', 'sse4a', 'avx512bw', 'clflushopt', 'xsave',
+                       'avx512vbmi2', '64bit', 'avx512vl', 'serialize', 'hreset',
+                       'invpcid', 'avx512cd', 'avx', 'vaes', 'avx512bf16', 'cx8',
+                       'fma', 'rtm', 'bmi', 'enqcmd', 'rdrnd', 'mwaitx', 'sse4.1',
+                       'sse4.2', 'avx2', 'fxsr', 'wbnoinvd', 'sse', 'lzcnt', 'pclmul',
+                       'prefetchwt1', 'f16c', 'ssse3', 'sgx', 'shstk', 'cmov',
+                       'avx512vbmi', 'amxint8', 'movbe', 'avx512vp2intersect',
+                       'xsaveopt', 'avx512dq', 'sse2', 'adx', 'sse3'],
             (11, 8): ['tsxldtrk', 'amx-tile', 'amx-bf16', 'serialize', 'amx-int8',
                       'avx512vp2intersect', 'tsxldtrk', 'amx-tile', 'amx-bf16',
                       'serialize', 'amx-int8', 'avx512vp2intersect', 'tsxldtrk',
@@ -231,7 +247,6 @@ def post_lowering(self, mod, library):
 # ---------------------------------------------------------------------------
 # Code generation methods
 
-
 @contextmanager
 def replace_numba_internals_hack():
     # Hackish solution to prevent numba from calling _ensure_finalize. See issue #87

rbc/libfuncs.py

@@ -282,6 +282,7 @@ def check(self, fname):
     experimental.vector.reduce.smin.* experimental.vector.reduce.umax.*
     experimental.vector.reduce.umin.* experimental.vector.reduce.fmax.*
     experimental.vector.reduce.fmin.*
+    experimental.noalias.scope.decl
     flt.rounds var.annotation ptr.annotation annotation
     codeview.annotation trap debugtrap stackprotector stackguard
     objectsize expect expect.with.probability assume ssa_copy

rbc/remotejit.py

@@ -3,17 +3,20 @@
 
 __all__ = ['RemoteJIT', 'Signature', 'Caller']
 
-import os
+import ctypes
 import inspect
+import os
 import warnings
-import ctypes
 from collections import defaultdict
+
+import llvmlite.binding as llvm
+
 from . import irtools, config
 from .errors import UnsupportedError
-from .typesystem import Type, get_signature
-from .thrift import Server, Dispatcher, dispatchermethod, Data, Client
-from .utils import get_local_ip, UNSPECIFIED, validate_devices
 from .targetinfo import TargetInfo
+from .thrift import Client, Data, Dispatcher, Server, dispatchermethod
+from .typesystem import Type, get_signature
+from .utils import UNSPECIFIED, get_local_ip, validate_devices
 
 
 def isfunctionlike(obj):
@@ -990,6 +993,7 @@ def targets(self) -> dict:
         target_info = TargetInfo.host()
         target_info.set('has_numba', True)
         target_info.set('has_cpython', True)
+        target_info.set('llvm_version', llvm.llvm_version_info)
         return dict(cpu=target_info.tojson())
 
     @dispatchermethod

rbc/stdlib/creation_functions.py

@@ -105,7 +105,7 @@ def _array_api_triu(x, k=0):
 
 
 @expose.implements('full')
-def _impl__full(shape, fill_value, dtype=None):
+def _impl_full(shape, fill_value, dtype=None):
     """
     Return a new array of given shape and type, filled with fill_value.
     """

rbc/tests/heavydb/test_heavydb.py

@@ -1,10 +1,11 @@
-import os
 import itertools
-import pytest
+import os
+
 import numpy as np
+import pytest
 
-from rbc.errors import UnsupportedError, HeavyDBServerError
-from rbc.tests import heavydb_fixture, assert_equal
+from rbc.errors import HeavyDBServerError, UnsupportedError
+from rbc.tests import assert_equal, heavydb_fixture
 from rbc.typesystem import Type
 
 rbc_heavydb = pytest.importorskip('rbc.heavydb')