pytorch/torch/csrc/cuda
Peter Bell 3d79ced8cf wrap_pybind_function: support member function pointers (#88932)
This updates `wrap_pybind_function` to use `invoke` and adds the
`invoke_traits` object which is analogous to `function_traits` but
for member functions it includes the class as an explicit argument.

To test this is working properly, I've also applied it to the
`CUDAGraph` binding code.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88932
Approved by: https://github.com/albanD
2022-11-14 18:47:34 +00:00
..
shared Check all CUDA API calls for errors in torch/ (#81560) 2022-10-28 00:40:48 +00:00
comm.cpp
comm.h
device_set.h
Event.cpp
Event.h
Graph.cpp wrap_pybind_function: support member function pointers (#88932) 2022-11-14 18:47:34 +00:00
memory_snapshot.cpp
memory_snapshot.h
Module.cpp [allocator] Introduce the abstract class CUDACachingAllocator (#87251) 2022-10-20 01:17:00 +00:00
Module.h
nccl.cpp Check all CUDA API calls for errors in torch/ (#81560) 2022-10-28 00:40:48 +00:00
nccl.h [allocator] Move getFreeMutex (#87237) 2022-10-19 18:00:40 +00:00
python_comm.cpp
python_comm.h
python_nccl.cpp
python_nccl.h
Stream.cpp
Stream.h
Tensor.cpp
THCP.h
utils.cpp