File tree Expand file tree Collapse file tree 2 files changed +15
-4
lines changed
Expand file tree Collapse file tree 2 files changed +15
-4
lines changed Original file line number Diff line number Diff line change @@ -12,8 +12,14 @@ enum TritonPluginResult {
1212 TP_GENERIC_FAILURE = 1 ,
1313};
1414};
15- #define TRITON_PLUGIN_API \
16- extern " C" __attribute__((visibility(" default" ))) TritonPluginResult
15+
16+ #if defined(_WIN32)
17+ #define EXPORT_FUNC __declspec (dllexport)
18+ #else
19+ #define EXPORT_FUNC __attribute__ ((visibility(" default" )))
20+ #endif
21+
22+ #define TRITON_PLUGIN_API extern " C" EXPORT_FUNC TritonPluginResult
1723
1824struct TritonPlugin {
1925 TritonPlugin () = delete ;
Original file line number Diff line number Diff line change 66#include " triton/Tools/PluginUtils.h"
77#include < unordered_map>
88
9- #define TRITON_PLUGIN_API \
10- extern " C" __attribute__((visibility(" default" ))) TritonPluginResult
9+ #if defined(_WIN32)
10+ #define EXPORT_FUNC __declspec (dllexport)
11+ #else
12+ #define EXPORT_FUNC __attribute__ ((visibility(" default" )))
13+ #endif
14+
15+ #define TRITON_PLUGIN_API extern " C" EXPORT_FUNC TritonPluginResult
1116
1217namespace mlir {
1318namespace triton {
You can’t perform that action at this time.
0 commit comments