Skip to content

Commit

Permalink
Fix syntax error
Browse files Browse the repository at this point in the history
  • Loading branch information
shendiaomo committed Nov 4, 2020
1 parent 2140b09 commit 1b2b2fa
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion cgotorch/parallel.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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<goModule>(go_module, f),
*input)
*input);
return nullptr;
} catch (const std::exception &e) {
return exception_str(e.what());
}
Expand Down

0 comments on commit 1b2b2fa

Please sign in to comment.