Edward Yang da4033d32a Make cudaHostRegister actually useful on cudart. (#45159) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45159 By default, pybind11 binds void* to be capsules. After a lot of Googling, I have concluded that this is not actually useful: you can't actually create a capsule from Python land, and our data_ptr() function returns an int, which means that the function is effectively unusable. It didn't help that we had no tests exercising it. I've replaced the void* with uintptr_t, so that we now accept int (and you can pass data_ptr() in directly). I'm not sure if we should make these functions accept ctypes types; unfortunately, pybind11 doesn't seem to have any easy way to do this. Fixes #43006 Also added cudaHostUnregister which was requested. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Test Plan: Imported from OSS Reviewed By: lw Differential Revision: D23849731 Pulled By: ezyang fbshipit-source-id: 8a79986f3aa9546abbd2a6a5828329ae90fd298f		2020-09-23 11:05:44 -07:00
..
cudart.cpp	Make cudaHostRegister actually useful on cudart. (#45159)	2020-09-23 11:05:44 -07:00
cudnn.cpp	Use compile time cudnn version if linking with it statically (#44402)	2020-09-09 11:33:41 -07:00
nvtx.cpp