mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
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 |
||
|---|---|---|
| .. | ||
| shared | ||
| comm.cpp | ||
| comm.h | ||
| device_set.h | ||
| Event.cpp | ||
| Event.h | ||
| Graph.cpp | ||
| memory_snapshot.cpp | ||
| memory_snapshot.h | ||
| Module.cpp | ||
| Module.h | ||
| nccl.cpp | ||
| nccl.h | ||
| python_comm.cpp | ||
| python_comm.h | ||
| python_nccl.cpp | ||
| python_nccl.h | ||
| Stream.cpp | ||
| Stream.h | ||
| Tensor.cpp | ||
| THCP.h | ||
| utils.cpp | ||