extension-ffi
extension-ffi copied to clipboard
C++ testing and validation
How can I be in c + + compiler environment, testing and validation. there the ‘THFloatTensor_data’ is not defined, when i use the code as follow:
int my_lib_add_forward(THFloatTensor *input1, THFloatTensor *input2,
THFloatTensor *output)
{
float * data_flat = THFloatTensor_data(input1);
}
Hi, you can try to write something like
include<torch/torch.h>
int my_lib_add_forward(at::Tensor *input1, at::Tensor *input2, at::Tensor *output){ auto tmp = input1->contiguous(); float *data_flat = tmp.data
(); }
good luck