Skip to content

Commit ed1efc4

Browse files
committed
[Intel] Fix Windows build after 'd907aed'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 7c28a58 commit ed1efc4

File tree

2 files changed

+15
-4
lines changed

2 files changed

+15
-4
lines changed

include/triton/Tools/PluginUtils.h

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff 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

1824
struct TritonPlugin {
1925
TritonPlugin() = delete;

lib/Plugins/TritonPlugin.cpp

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,13 @@
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

1217
namespace mlir {
1318
namespace triton {

0 commit comments

Comments
 (0)