8000 Add GPU tests for DLPack support by leofang · Pull Request #59 · mpi4py/mpi4py · GitHub
[go: up one dir, main page]
Skip to content

Add GPU tests for DLPack support #59

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 9 commits into from
Closed

Add GPU tests for DLPack support #59

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
9893755
dlpack: Add support for DLPack
dalcinl Jun 4, 2021
7338c11
dlpack: Add exhaustive tests for DLPack interface
dalcinl Jun 20, 2021
4acb6bd
simplify a bit DLPack support and add comment
leofang Jul 1, 2021
237359e
add DLPackCuPy arrayimpl
leofang Jul 1, 2021
47669e0
test if __dlpack_device__ is defined
leofang Jul 1, 2021
442fcb9
rename GPUBuf to CAIBuf for clarity
leofang Jul 1, 2021
79f64d5
rename DLPackBuf to DLPackCPUBuf for clarity
leofang Jul 1, 2021
0780e63
add simple message tests via DLPack GPU support
leofang Jul 1, 2021
0d955ae
fix DLPackCuPy tests
leofang Jul 2, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
test if __dlpack_device__ is defined
  • Loading branch information
@leofang
leofang committed Jul 1, 2021
commit 47669e0ee7b4d14f776e350730dd3a6c63e800f0
2 changes: 1 addition & 1 deletion test/test_msgspec.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -523,7 +523,7 @@ class TestMessageDLPackBuf(unittest.TestCase):
def testDevice(self):
buf = DLPackBuf('i', [0,1,2,3])
buf.__dlpack_device__ = None
MPI.Get_address(buf)
self.assertRaises(TypeError, MPI.Get_address, buf)
buf.__dlpack_device__ = lambda: None
self.assertRaises(TypeError, MPI.Get_address, buf)
buf.__dlpack_device__ = lambda: (None, 0)
Expand Down
0