diff --git a/cgotorch/parallel.cc b/cgotorch/parallel.cc index 18dc634..584630a 100644 --- a/cgotorch/parallel.cc +++ b/cgotorch/parallel.cc @@ -31,7 +31,8 @@ const char *DataParallel(char *go_module, void *f, Tensor input, "invalid memory address or nil pointer dereference of input tensor"); } torch::nn::parallel::data_parallel(std::make_shared(go_module, f), - *input) + *input); + return nullptr; } catch (const std::exception &e) { return exception_str(e.what()); }