numba-cuda 0.21.1__cp313-cp313-win_amd64.whl → 0.24.0__cp313-cp313-win_amd64.whl

numba_cuda/numba/cuda/cext/_devicearray.cpp

@@ -1,159 +0,0 @@
-// SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-// SPDX-License-Identifier: BSD-2-Clause
-
-/* This file contains the base class implementation for all device arrays. The
- * base class is implemented in C so that computing typecodes for device arrays
- * can be implemented efficiently. */
-
-#include "_pymodule.h"
-
-
-/* Include _devicearray., but make sure we don't get the definitions intended
- * for consumers of the Device Array API.
- */
-#define NUMBA_IN_DEVICEARRAY_CPP_
-#include "_devicearray.h"
-
-/* DeviceArray PyObject implementation. Note that adding more members here is
- * presently prohibited because mapped and managed arrays derive from both
- * DeviceArray and NumPy's ndarray, which is also a C extension class - the
- * layout of the object cannot be resolved if this class also has members beyond
- * PyObject_HEAD. */
-class DeviceArray {
-    PyObject_HEAD
-};
-
-/* Trivial traversal - DeviceArray instances own nothing. */
-static int
-DeviceArray_traverse(DeviceArray *self, visitproc visit, void *arg)
-{
-    return 0;
-}
-
-/* Trivial clear of all references - DeviceArray instances own nothing. */
-static int
-DeviceArray_clear(DeviceArray *self)
-{
-    return 0;
-}
-
-/* The _devicearray.DeviceArray type */
-PyTypeObject DeviceArrayType = {
-    PyVarObject_HEAD_INIT(NULL, 0)
-    "_devicearray.DeviceArray",                  /* tp_name */
-    sizeof(DeviceArray),                         /* tp_basicsize */
-    0,                                           /* tp_itemsize */
-    0,                                           /* tp_dealloc */
-    0,                                           /* tp_vectorcall_offset */
-    0,                                           /* tp_getattr */
-    0,                                           /* tp_setattr */
-    0,                                           /* tp_as_async */
-    0,                                           /* tp_repr */
-    0,                                           /* tp_as_number */
-    0,                                           /* tp_as_sequence */
-    0,                                           /* tp_as_mapping */
-    0,                                           /* tp_hash */
-    0,                                           /* tp_call*/
-    0,                                           /* tp_str*/
-    0,                                           /* tp_getattro*/
-    0,                                           /* tp_setattro*/
-    0,                                           /* tp_as_buffer*/
-    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,
-                                                 /* tp_flags*/
-    "DeviceArray object",                        /* tp_doc */
-    (traverseproc) DeviceArray_traverse,         /* tp_traverse */
-    (inquiry) DeviceArray_clear,                 /* tp_clear */
-    0,                                           /* tp_richcompare */
-    0,                                           /* tp_weaklistoffset */
-    0,                                           /* tp_iter */
-    0,                                           /* tp_iternext */
-    0,                                           /* tp_methods */
-    0,                                           /* tp_members */
-    0,                                           /* tp_getset */
-    0,                                           /* tp_base */
-    0,                                           /* tp_dict */
-    0,                                           /* tp_descr_get */
-    0,                                           /* tp_descr_set */
-    0,                                           /* tp_dictoffset */
-    0,                                           /* tp_init */
-    0,                                           /* tp_alloc */
-    0,                                           /* tp_new */
-    0,                                           /* tp_free */
-    0,                                           /* tp_is_gc */
-    0,                                           /* tp_bases */
-    0,                                           /* tp_mro */
-    0,                                           /* tp_cache */
-    0,                                           /* tp_subclasses */
-    0,                                           /* tp_weaklist */
-    0,                                           /* tp_del */
-    0,                                           /* tp_version_tag */
-    0,                                           /* tp_finalize */
-    0,                                           /* tp_vectorcall */
-#if (PY_MAJOR_VERSION == 3) && (PY_MINOR_VERSION == 12)
-/* This was introduced first in 3.12
- * https://github.com/python/cpython/issues/91051
- */
-    0,                                           /* tp_watched */
-#endif
-
-/* WARNING: Do not remove this, only modify it! It is a version guard to
- * act as a reminder to update this struct on Python version update! */
-#if (PY_MAJOR_VERSION == 3)
-#if ! (NB_SUPPORTED_PYTHON_MINOR)
-#error "Python minor version is not supported."
-#endif
-#else
-#error "Python major version is not supported."
-#endif
-/* END WARNING*/
-};
-
-/* CUDA device array C API */
-static void *_DeviceArray_API[1] = {
-    (void*)&DeviceArrayType
-};
-
-MOD_INIT(_devicearray) {
-    PyObject *m = nullptr;
-    PyObject *d = nullptr;
-    PyObject *c_api = nullptr;
-    int error = 0;
-
-    MOD_DEF(m, "_devicearray", "No docs", NULL)
-    if (m == NULL)
-        goto error_occurred;
-
-    c_api = PyCapsule_New((void *)_DeviceArray_API, NUMBA_DEVICEARRAY_IMPORT_NAME "._DEVICEARRAY_API", NULL);
-    if (c_api == NULL)
-        goto error_occurred;
-
-    DeviceArrayType.tp_new = PyType_GenericNew;
-    if (PyType_Ready(&DeviceArrayType) < 0)
-        goto error_occurred;
-
-    Py_INCREF(&DeviceArrayType);
-    error = PyModule_AddObject(m, "DeviceArray", (PyObject*)(&DeviceArrayType));
-    if (error)
-        goto error_occurred;
-
-    d = PyModule_GetDict(m);
-    if (d == NULL)
-        goto error_occurred;
-
-    error = PyDict_SetItemString(d, "_DEVICEARRAY_API", c_api);
-    /* Decref and set c_api to NULL, Py_XDECREF in error_occurred will have no
-     * effect. */
-    Py_CLEAR(c_api);
-
-    if (error)
-        goto error_occurred;
-
-    return MOD_SUCCESS_VAL(m);
-
-error_occurred:
-    Py_XDECREF(m);
-    Py_XDECREF(c_api);
-    Py_XDECREF((PyObject*)&DeviceArrayType);
-
-    return MOD_ERROR_VAL;
-}

numba_cuda/numba/cuda/cext/_devicearray.h

@@ -1,29 +0,0 @@
-// SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-// SPDX-License-Identifier: BSD-2-Clause
-
-#ifndef NUMBA_DEVICEARRAY_H_
-#define NUMBA_DEVICEARRAY_H_
-
-#ifdef __cplusplus
-    extern "C" {
-#endif
-
-#define NUMBA_DEVICEARRAY_IMPORT_NAME "numba.cuda.cext._devicearray"
-/* These definitions should only be used by consumers of the Device Array API.
- * Consumers access the API through the opaque pointer stored in
- * _devicearray._DEVICEARRAY_API.  We don't want these definitions in
- * _devicearray.cpp itself because they would conflict with the actual
- * implementations there.
- */
-#ifndef NUMBA_IN_DEVICEARRAY_CPP_
-
-    extern void **DeviceArray_API;
-    #define DeviceArrayType (*(PyTypeObject*)DeviceArray_API[0])
-
-#endif /* ndef NUMBA_IN_DEVICEARRAY_CPP */
-
-#ifdef __cplusplus
-    }
-#endif
-
-#endif  /* NUMBA_DEVICEARRAY_H_ */

numba_cuda/numba/cuda/intrinsic_wrapper.py

@@ -1,41 +0,0 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: BSD-2-Clause
-
-from .decorators import jit
-import numba
-
-
-@jit(device=True)
-def all_sync(mask, predicate):
-    """
-    If for all threads in the masked warp the predicate is true, then
-    a non-zero value is returned, otherwise 0 is returned.
-    """
-    return numba.cuda.vote_sync_intrinsic(mask, 0, predicate)[1]
-
-
-@jit(device=True)
-def any_sync(mask, predicate):
-    """
-    If for any thread in the masked warp the predicate is true, then
-    a non-zero value is returned, otherwise 0 is returned.
-    """
-    return numba.cuda.vote_sync_intrinsic(mask, 1, predicate)[1]
-
-
-@jit(device=True)
-def eq_sync(mask, predicate):
-    """
-    If for all threads in the masked warp the boolean predicate is the same,
-    then a non-zero value is returned, otherwise 0 is returned.
-    """
-    return numba.cuda.vote_sync_intrinsic(mask, 2, predicate)[1]
-
-
-@jit(device=True)
-def ballot_sync(mask, predicate):
-    """
-    Returns a mask of all threads in the warp whose predicate is true,
-    and are within the given mask.
-    """
-    return numba.cuda.vote_sync_intrinsic(mask, 3, predicate)[0]