diff --git a/src/ctorch.h b/src/ctorch.h index 733d26dc..9b45102d 100644 --- a/src/ctorch.h +++ b/src/ctorch.h @@ -7,6 +7,8 @@ #define EXPORT_C #endif +#include + // Opaque pointer type alias for torch::jit::script::Module class typedef void *torch_jit_script_module_t;