Home
last modified time | relevance | path

Searched refs:__cuda_array_interface__ (Results 1 – 25 of 29) sorted by relevance

12

/dports/net/py-mpi4py/mpi4py-3.1.3/test/
H A Dtest_msgspec.py138 self.__cuda_array_interface__ = dict(
700 rmsg.__cuda_array_interface__ = None
706 rmsg.__cuda_array_interface__ = dict()
713 rmsg.__cuda_array_interface__ = items
719 del rmsg.__cuda_array_interface__['data']
731 rmsg.__cuda_array_interface__['data'] = 0
740 rmsg.__cuda_array_interface__['data'] = ( )
773 del rmsg.__cuda_array_interface__['shape']
785 rmsg.__cuda_array_interface__['shape'] = 3
818 del rmsg.__cuda_array_interface__['descr']
[all …]
H A Darrayimpl.py307 self.__cuda_array_interface__ = dict(
373 return self.array.__cuda_array_interface__['data'][0]
479 return self.array.__cuda_array_interface__['data'][0]
/dports/devel/py-numba/numba-0.51.2/numba/cuda/tests/cudapy/
H A Dtest_cuda_array_interface.py179 c_ai = c_arr.__cuda_array_interface__
182 c_ai_sliced = c_arr[::-1].__cuda_array_interface__
210 masked_cuda_array_interface = c_arr.__cuda_array_interface__.copy()
221 self.assertEqual(c_arr.__cuda_array_interface__['data'][0], 0)
237 self.assertEqual(c_arr.__cuda_array_interface__['strides'], None)
241 self.assertNotEqual(c_arr.__cuda_array_interface__['strides'], None)
246 face = devarray.__cuda_array_interface__
/dports/net/py-mpi4py/mpi4py-3.1.3/src/mpi4py/MPI/
H A Dascaibuf.pxi55 try: return <bint>hasattr(obj, '__cuda_array_interface__')
74 cuda_array_interface = obj.__cuda_array_interface__
97 "__cuda_array_interface__: "
102 "__cuda_array_interface__: "
110 "__cuda_array_interface__: "
116 b"__cuda_array_interface__: "
/dports/devel/py-numba/numba-0.51.2/numba/cuda/
H A Dtesting.py121 self.__cuda_array_interface__ = arr.__cuda_array_interface__
H A Dapi.py66 return from_cuda_array_interface(obj.__cuda_array_interface__,
/dports/misc/py-xgboost/xgboost-1.5.1/python-package/xgboost/
H A Ddata.py475 interface = cat_codes[0].__cuda_array_interface__
477 interface = data.__cuda_array_interface__
479 interface["mask"] = interface["mask"].__cuda_array_interface__
485 interface = codes.__cuda_array_interface__
487 interface = data[col].__cuda_array_interface__
489 interface["mask"] = interface["mask"].__cuda_array_interface__
865 interface = bytes(json.dumps([data.__cuda_array_interface__],
873 interface = bytes(json.dumps([data.__cuda_array_interface__],
882 interface = bytes(json.dumps([data.__cuda_array_interface__],
H A Dcore.py243 interface = data.__cuda_array_interface__
245 interface["mask"] = interface["mask"].__cuda_array_interface__
1108 interface = data.__cuda_array_interface__
/dports/misc/xgboost/xgboost-1.5.1/python-package/xgboost/
H A Ddata.py475 interface = cat_codes[0].__cuda_array_interface__
477 interface = data.__cuda_array_interface__
479 interface["mask"] = interface["mask"].__cuda_array_interface__
485 interface = codes.__cuda_array_interface__
487 interface = data[col].__cuda_array_interface__
489 interface["mask"] = interface["mask"].__cuda_array_interface__
865 interface = bytes(json.dumps([data.__cuda_array_interface__],
873 interface = bytes(json.dumps([data.__cuda_array_interface__],
882 interface = bytes(json.dumps([data.__cuda_array_interface__],
H A Dcore.py243 interface = data.__cuda_array_interface__
245 interface["mask"] = interface["mask"].__cuda_array_interface__
1108 interface = data.__cuda_array_interface__
/dports/devel/py-distributed/distributed-2021.11.2/distributed/protocol/
H A Drmm.py14 header = x.__cuda_array_interface__.copy()
H A Dnumba.py24 header = x.__cuda_array_interface__.copy()
H A Dcupy.py23 header = x.__cuda_array_interface__.copy()
/dports/devel/py-numba/numba-0.51.2/docs/source/cuda/
H A Dcuda_array_interface.rst23 The ``__cuda_array_interface__`` attribute returns a dictionary (``dict``)
68 - **mask**: ``None`` or object exposing the ``__cuda_array_interface__``
83 Obtaining the value of the ``__cuda_array_interface__`` property of any object
/dports/www/grafana8/grafana-8.3.6/vendor/github.com/apache/arrow/python/pyarrow/tests/
H A Dtest_cuda_numba_interop.py148 def __cuda_array_interface__(self): member in test_from_object.MyObj
149 return self.darr.__cuda_array_interface__
/dports/databases/arrow/apache-arrow-6.0.1/python/pyarrow/tests/
H A Dtest_cuda_numba_interop.py148 def __cuda_array_interface__(self): member in test_from_object.MyObj
149 return self.darr.__cuda_array_interface__
/dports/security/vault/vault-1.8.2/vendor/github.com/apache/arrow/python/pyarrow/tests/
H A Dtest_cuda_numba_interop.py148 def __cuda_array_interface__(self): member in test_from_object.MyObj
149 return self.darr.__cuda_array_interface__
/dports/net/py-mpi4py/mpi4py-3.1.3/docs/source/usrman/
H A Dtutorial.rst107 ``__cuda_array_interface__`` attribute that are compliant with the
406 assert hasattr(sendbuf, '__cuda_array_interface__')
407 assert hasattr(recvbuf, '__cuda_array_interface__')
/dports/devel/py-numba/numba-0.51.2/numba/cuda/cudadrv/
H A Ddevicearray.py119 def __cuda_array_interface__(self): member in DeviceNDArrayBase
/dports/www/grafana8/grafana-8.3.6/vendor/github.com/apache/arrow/python/pyarrow/
H A D_cuda.pyx316 elif hasattr(obj, '__cuda_array_interface__'):
317 desc = obj.__cuda_array_interface__
/dports/databases/arrow/apache-arrow-6.0.1/python/pyarrow/
H A D_cuda.pyx317 elif hasattr(obj, '__cuda_array_interface__'):
318 desc = obj.__cuda_array_interface__
/dports/security/vault/vault-1.8.2/vendor/github.com/apache/arrow/python/pyarrow/
H A D_cuda.pyx316 elif hasattr(obj, '__cuda_array_interface__'):
317 desc = obj.__cuda_array_interface__
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py1163 def __cuda_array_interface__(self): member in _DeviceArray
1164 return _force(self).device_buffer.__cuda_array_interface__
/dports/devel/py-numba/numba-0.51.2/
H A DCHANGE_LOG589 * Leo Fang added a list of libraries that support ``__cuda_array_interface__``
594 * Mads R. B. Kristensen fixed an issue with ``__cuda_array_interface__`` not
710 * PR #5104: Add a list of libraries that support __cuda_array_interface__
713 * PR #5189: __cuda_array_interface__ not requiring context
826 * John Kirkham added a clarification to the ``__cuda_array_interface__``
1379 * Keith Kraus extended the ``__cuda_array_interface__`` with an optional mask
1436 * PR #4199: Extend `__cuda_array_interface__` with optional mask attribute,
1946 * PR #3335: Fix memory management of __cuda_array_interface__ views.
2186 * For CUDA GPU support, we added a `__cuda_array_interface__` following the
2228 * PR #2860: __cuda_array_interface__
/dports/net/py-mpi4py/mpi4py-3.1.3/
H A DCHANGES.rst67 interchange mechanism (`link <DIM_>`_) and the ``__cuda_array_interface__``

12