How to get multiple Tensors returned from CUDA extension

Hi there,

I want to transfer two tensors into my self-defined CUDA extension and return them back after some operations. However, as C++ only support to return one single variable, this seems to be a bit hard to implement. I tried to use boost tuple to combine them, but got an error shown below:

undefined symbol: _ZN5boost6python23throw_error_already_setEv

I also tried to use data_ptr() to get the address of the tensor as an int value, however, I don’t know how to cast the int value into a Tensor pointer in my cuda program. Do you know how to deal with that?


Internally, using variable_list = std::vector<Variable>; is used.
I am not sure if pybind will convert that properly though :confused: