Home
last modified time | relevance | path

Searched refs:dlpack (Results 1 – 25 of 106) sorted by relevance

12345

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_dlpack.py24 def from_dlpack_old(dlpack): argument
30 dlpack = ctypes.py_object(dlpack)
31 assert ctypes.pythonapi.PyCapsule_IsValid(dlpack, _c_str_dltensor), ValueError(
33 … dlpack_handle = ctypes.c_void_p(ctypes.pythonapi.PyCapsule_GetPointer(dlpack, _c_str_dltensor))
36 ctypes.pythonapi.PyCapsule_SetName(dlpack, _c_str_used_dltensor)
38 ctypes.pythonapi.PyCapsule_SetDestructor(dlpack, None)
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/python/unittest/
H A Dtest_dlpack.py24 def from_dlpack_old(dlpack): argument
30 dlpack = ctypes.py_object(dlpack)
31 assert ctypes.pythonapi.PyCapsule_IsValid(dlpack, _c_str_dltensor), ValueError(
33 … dlpack_handle = ctypes.c_void_p(ctypes.pythonapi.PyCapsule_GetPointer(dlpack, _c_str_dltensor))
36 ctypes.pythonapi.PyCapsule_SetName(dlpack, _c_str_used_dltensor)
38 ctypes.pythonapi.PyCapsule_SetDestructor(dlpack, None)
/dports/net/py-mpi4py/mpi4py-3.1.3/src/mpi4py/MPI/
H A Dasdlpack.pxi3 # From dlpack.h (as of v0.6)
89 raise BufferError("dlpack: number of dimensions is negative")
91 raise BufferError("dlpack: shape is NULL")
94 raise BufferError("dlpack: shape item is negative")
105 raise BufferError("dlpack: buffer is not contiguous")
173 cdef object dlpack
186 dlpack = obj.__dlpack__
189 raise NotImplementedError("dlpack: missing support")
193 capsule = dlpack()
195 capsule = dlpack(stream=-1)
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/
H A D.gitmodules4 [submodule "dlpack"]
5 path = 3rdparty/dlpack
6 url = https://github.com/dmlc/dlpack
/dports/misc/py-tvm/incubator-tvm-0.6.1/
H A D.gitmodules4 [submodule "dlpack"]
5 path = 3rdparty/dlpack
6 url = https://github.com/dmlc/dlpack
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/
H A D.gitmodules4 [submodule "dlpack"]
5 path = 3rdparty/dlpack
6 url = https://github.com/dmlc/dlpack
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/
H A Dtest_dlpack.py20 from tvm.contrib.dlpack import to_pytorch_func
30 import torch.utils.dlpack
33 tvm_x = tvm.nd.from_dlpack(torch.utils.dlpack.to_dlpack(x))
38 torch.utils.dlpack.from_dlpack(y.to_dlpack()).numpy(), tvm_x.asnumpy()
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/contrib/
H A Dtest_dlpack.py19 from tvm.contrib.dlpack import to_pytorch_func
28 import torch.utils.dlpack
31 tvm_x = tvm.nd.from_dlpack(torch.utils.dlpack.to_dlpack(x))
35 … np.testing.assert_equal(torch.utils.dlpack.from_dlpack(y.to_dlpack()).numpy(), tvm_x.asnumpy())
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/contrib/
H A Dtest_dlpack.py19 from tvm.contrib.dlpack import to_pytorch_func
28 import torch.utils.dlpack
31 tvm_x = tvm.nd.from_dlpack(torch.utils.dlpack.to_dlpack(x))
35 … np.testing.assert_equal(torch.utils.dlpack.from_dlpack(y.to_dlpack()).numpy(), tvm_x.asnumpy())
/dports/misc/mxnet/incubator-mxnet-1.9.0/
H A D.gitmodules7 [submodule "3rdparty/dlpack"]
8 path = 3rdparty/dlpack
9 url = https://github.com/dmlc/dlpack
H A D.licenserc.yaml78 - 'include/dlpack' # symlink to 3rdparty/dlpack/include/dlpack
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/
H A D.gitmodules7 [submodule "3rdparty/dlpack"]
8 path = 3rdparty/dlpack
9 url = https://github.com/dmlc/dlpack
H A D.licenserc.yaml78 - 'include/dlpack' # symlink to 3rdparty/dlpack/include/dlpack
/dports/net/py-mpi4py/mpi4py-3.1.3/test/
H A Darrayimpl.py244 import dlpackimpl as dlpack namespace
246 dlpack = None variable
251 return (dlpack.DLDeviceType.kDLCPU, 0)
255 capsule = dlpack.make_py_capsule(self.array)
262 if dlpack is not None and array is not None:
273 if dlpack is not None and numpy is not None:
414 self.dev_type = dlpack.DLDeviceType.kDLROCM
416 self.dev_type = dlpack.DLDeviceType.kDLCUDA
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/contrib/
H A Ddlpack.py58 import torch.utils.dlpack
59 return convert_func(tvm_func, torch.Tensor, torch.utils.dlpack.to_dlpack)
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/contrib/
H A Ddlpack.py58 import torch.utils.dlpack
59 return convert_func(tvm_func, torch.Tensor, torch.utils.dlpack.to_dlpack)
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/
H A Ddlpack.py63 import torch.utils.dlpack
65 return convert_func(tvm_func, torch.Tensor, torch.utils.dlpack.to_dlpack)
/dports/math/py-jax/jax-0.2.9/jax/_src/
H A Ddlpack.py50 def from_dlpack(dlpack, backend=None): argument
63 buf = xla_client._xla.dlpack_managed_tensor_to_buffer(dlpack, client)
/dports/misc/tvm/incubator-tvm-0.6.1/3rdparty/dlpack/
H A DREADME.md3 [![Build Status](https://travis-ci.org/dmlc/dlpack.svg?branch=master)](https://travis-ci.org/dmlc/d…
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/dlpack/
H A DREADME.md3 [![Build Status](https://travis-ci.org/dmlc/dlpack.svg?branch=master)](https://travis-ci.org/dmlc/d…
/dports/misc/py-tvm/incubator-tvm-0.6.1/3rdparty/dlpack/
H A DREADME.md3 [![Build Status](https://travis-ci.org/dmlc/dlpack.svg?branch=master)](https://travis-ci.org/dmlc/d…
/dports/misc/tvm/incubator-tvm-0.6.1/apps/sgx/
H A DMakefile42 -I$(TVM_DIR)/dlpack/include \
65 -I$(TVM_DIR)/dlpack/include \
/dports/misc/py-tvm/incubator-tvm-0.6.1/apps/sgx/
H A DMakefile42 -I$(TVM_DIR)/dlpack/include \
65 -I$(TVM_DIR)/dlpack/include \
/dports/misc/tvm/incubator-tvm-0.6.1/3rdparty/dlpack/contrib/
H A Dmock_main.cc6 dlpack::DLTContainer c; in main()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/dlpack/contrib/
H A Dmock_main.cc6 dlpack::DLTContainer c; in main()

12345