/usr/local/lib64/python3.6/site-packages/torch/include/THC
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/THC/THCAllocator.h (370B)
#ifndef THC_ALLOCATOR_INC
#define THC_ALLOCATOR_INC
#include
// IPC doesn't support (re)allocation
class TORCH_CUDA_CPP_API THCIpcDeleter {
public:
THCIpcDeleter(std::shared_ptr basePtr);
~THCIpcDeleter();
static at::DataPtr makeDataPtr(std::shared_ptr basePtr, void* data);
private:
std::shared_ptr basePtr_;
};
#endif