Skip to content

Commit 40561cf

Browse files
committed
add pti_view.lib
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 8920194 commit 40561cf

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

third_party/proton/csrc/lib/Driver/GPU/XpuptiApi.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,11 @@ namespace xpupti {
88

99
struct ExternLibXpupti : public ExternLibBase {
1010
using RetType = pti_result;
11+
#if defined(_WIN32)
1112
static constexpr const char *name = "libpti_view.so";
13+
#else
14+
static constexpr const char *name = "pti_view.lib";
15+
#endif
1216
static constexpr const char *defaultDir = "";
1317
static constexpr const char *pathEnv = "TRITON_XPUPTI_LIB_PATH";
1418
static constexpr RetType success = PTI_SUCCESS;

third_party/proton/proton/profile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def _select_backend() -> str:
2727
try:
2828
if (files := importlib.metadata.files('intel-pti')) is not None:
2929
for f in files:
30-
if 'libpti_view.so' in f.name:
30+
if any(map(lambda el: el in f.name, ('libpti_view.so', 'pti_view.lib'))): # pylint: disable=W0640
3131
os.environ["TRITON_XPUPTI_LIB_PATH"] = str(pathlib.Path(f.locate()).parent.resolve())
3232
break
3333
except importlib.metadata.PackageNotFoundError:

0 commit comments

Comments
 (0)