Skip to content

Commit de83a5b

Browse files
committed
Revert "[CMake] Disable TestSofieModels if PyTorch >= 2.9.0"
This reverts commit 5f3a91f.
1 parent b48ab8a commit de83a5b

File tree

1 file changed

+8
-18
lines changed

1 file changed

+8
-18
lines changed

tmva/sofie/test/CMakeLists.txt

Lines changed: 8 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -116,24 +116,14 @@ if (ROOT_TORCH_FOUND)
116116
configure_file(RecurrentModelGenerator.py RecurrentModelGenerator.py COPYONLY)
117117

118118
if (BLAS_FOUND)
119-
execute_process(COMMAND "${Python3_EXECUTABLE}" "-c"
120-
"import torch; print(getattr(torch, '__version__', 'unknown'))"
121-
RESULT_VARIABLE status
122-
OUTPUT_VARIABLE module_version
123-
OUTPUT_STRIP_TRAILING_WHITESPACE
124-
ERROR_QUIET)
125-
if(${status} EQUAL 0 AND ${module_version} VERSION_LESS 2.9.0)
126-
# Disable TestSofieModels which is failing on alma8, alma10, ubuntu2204, and ubuntu2404
127-
# with PyTorch 2.9.0
128-
ROOT_ADD_GTEST(TestSofieModels TestSofieModels.cxx
129-
LIBRARIES
130-
ROOTTMVASofie
131-
ROOTTMVASofieParser
132-
BLAS::BLAS
133-
INCLUDE_DIRS
134-
${CMAKE_CURRENT_BINARY_DIR}
135-
)
136-
endif()
119+
ROOT_ADD_GTEST(TestSofieModels TestSofieModels.cxx
120+
LIBRARIES
121+
ROOTTMVASofie
122+
ROOTTMVASofieParser
123+
BLAS::BLAS
124+
INCLUDE_DIRS
125+
${CMAKE_CURRENT_BINARY_DIR}
126+
)
137127
endif()
138128
endif()
139129

0 commit comments

Comments
 (0)