Fix local cpp tests after inplace build (NVIDIA#911)

Add top level path to search for .so in test build Signed-off-by: Kirthi Shankar Sivamani <[email protected]>
erhoo82 · Jun 12, 2024 · 78efc93 · 78efc93
1 parent dff1134
commit 78efc93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cpp/CMakeLists.txt b/tests/cpp/CMakeLists.txt
@@ -26,7 +26,7 @@ if(NOT DEFINED TE_LIB_PATH)
                     OUTPUT_VARIABLE TE_LIB_PATH)
 endif()
 
-find_library(TE_LIB NAMES transformer_engine PATHS "${TE_LIB_PATH}/transformer_engine" ENV TE_LIB_PATH REQUIRED)
+find_library(TE_LIB NAMES transformer_engine PATHS "${TE_LIB_PATH}/transformer_engine" ${TE_LIB_PATH} ENV TE_LIB_PATH REQUIRED)
 
 message(STATUS "Found transformer_engine library: ${TE_LIB}")
 include_directories(../../transformer_engine/common/include)