/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/python/unittest/ |
H A D | test_dlpack.py | 24 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 D | test_dlpack.py | 24 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 D | asdlpack.pxi | 3 # 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 | .gitmodules | 4 [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 | .gitmodules | 4 [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 | .gitmodules | 4 [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 D | test_dlpack.py | 20 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 D | test_dlpack.py | 19 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 D | test_dlpack.py | 19 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 | .gitmodules | 7 [submodule "3rdparty/dlpack"] 8 path = 3rdparty/dlpack 9 url = https://github.com/dmlc/dlpack
|
H A D | .licenserc.yaml | 78 - 'include/dlpack' # symlink to 3rdparty/dlpack/include/dlpack
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/ |
H A D | .gitmodules | 7 [submodule "3rdparty/dlpack"] 8 path = 3rdparty/dlpack 9 url = https://github.com/dmlc/dlpack
|
H A D | .licenserc.yaml | 78 - 'include/dlpack' # symlink to 3rdparty/dlpack/include/dlpack
|
/dports/net/py-mpi4py/mpi4py-3.1.3/test/ |
H A D | arrayimpl.py | 244 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 D | dlpack.py | 58 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 D | dlpack.py | 58 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 D | dlpack.py | 63 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 D | dlpack.py | 50 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 D | README.md | 3 [![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 D | README.md | 3 [![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 D | README.md | 3 [![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 D | Makefile | 42 -I$(TVM_DIR)/dlpack/include \ 65 -I$(TVM_DIR)/dlpack/include \
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/apps/sgx/ |
H A D | Makefile | 42 -I$(TVM_DIR)/dlpack/include \ 65 -I$(TVM_DIR)/dlpack/include \
|
/dports/misc/tvm/incubator-tvm-0.6.1/3rdparty/dlpack/contrib/ |
H A D | mock_main.cc | 6 dlpack::DLTContainer c; in main()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/dlpack/contrib/ |
H A D | mock_main.cc | 6 dlpack::DLTContainer c; in main()
|