Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add ifdef __cplusplus to wrapper header file

Browse Source
This commit is contained in:
Travis Brady 2014-11-14 15:48:13 -06:00
parent b2850ae0f9
commit 42712988af
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
wrapper/xgboost_wrapper.h
View File

@ -6,7 +6,11 @@
* \brief a C style wrapper of xgboost * \brief a C style wrapper of xgboost
* can be used to create wrapper of other languages * can be used to create wrapper of other languages
*/ */
#ifdef __cplusplus
#include <cstdio> #include <cstdio>
#else
#include <stdio.h>
#endif
#ifdef _MSC_VER #ifdef _MSC_VER
#define XGB_DLL __declspec(dllexport) #define XGB_DLL __declspec(dllexport)
#else #else
@ -15,8 +19,9 @@
// manually define unsign long // manually define unsign long
typedef unsigned long bst_ulong; typedef unsigned long bst_ulong;
#ifdef __cplusplus
extern "C" { extern "C" {
#endif
/*! /*!
* \brief load a data matrix * \brief load a data matrix
* \return a loaded data matrix * \return a loaded data matrix
@ -205,5 +210,7 @@ extern "C" {
*/ */
XGB_DLL const char **XGBoosterDumpModel(void *handle, const char *fmap, XGB_DLL const char **XGBoosterDumpModel(void *handle, const char *fmap,
bst_ulong *out_len); bst_ulong *out_len);
#ifdef __cplusplus
} }
#endif
#endif // XGBOOST_WRAPPER_H_ #endif // XGBOOST_WRAPPER_H_