Skip to content

Commit bdd5458

Browse files
authored
Fix Python wheel build args without perf analyzer (#98)
1 parent 0300e34 commit bdd5458

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/python/library/CMakeLists.txt

+4-1
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,9 @@ if (NOT WIN32)
106106
)
107107
endif() # TRITON_ENABLE_GPU
108108

109+
if (${TRITON_ENABLE_PERF_ANALYZER})
110+
set(perf_analyzer_arg --perf-analyzer ${CMAKE_INSTALL_PREFIX}/bin/perf_analyzer)
111+
endif()
109112
set(linux_wheel_stamp_file "linux_stamp.whl")
110113
add_custom_command(
111114
OUTPUT "${linux_wheel_stamp_file}"
@@ -114,7 +117,7 @@ if (NOT WIN32)
114117
"${CMAKE_CURRENT_SOURCE_DIR}/build_wheel.py"
115118
--dest-dir "${CMAKE_CURRENT_BINARY_DIR}/linux"
116119
--linux
117-
--perf-analyzer ${CMAKE_INSTALL_PREFIX}/bin/perf_analyzer
120+
${perf_analyzer_arg}
118121
DEPENDS ${LINUX_WHEEL_DEPENDS}
119122
)
120123

0 commit comments

Comments
 (0)