Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 18 additions & 8 deletions tmva/sofie/test/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -116,14 +116,24 @@ if (ROOT_TORCH_FOUND)
configure_file(RecurrentModelGenerator.py RecurrentModelGenerator.py COPYONLY)

if (BLAS_FOUND)
ROOT_ADD_GTEST(TestSofieModels TestSofieModels.cxx
LIBRARIES
ROOTTMVASofie
ROOTTMVASofieParser
BLAS::BLAS
INCLUDE_DIRS
${CMAKE_CURRENT_BINARY_DIR}
)
execute_process(COMMAND "${Python3_EXECUTABLE}" "-c"
"import torch; print(getattr(torch, '__version__', 'unknown'))"
RESULT_VARIABLE status
OUTPUT_VARIABLE module_version
OUTPUT_STRIP_TRAILING_WHITESPACE
ERROR_QUIET)
if(${status} EQUAL 0 AND ${module_version} VERSION_LESS 2.9.0)
# Disable TestSofieModels which is failing on alma8, alma10, ubuntu2204, and ubuntu2404
# with PyTorch 2.9.0
ROOT_ADD_GTEST(TestSofieModels TestSofieModels.cxx
LIBRARIES
ROOTTMVASofie
ROOTTMVASofieParser
BLAS::BLAS
INCLUDE_DIRS
${CMAKE_CURRENT_BINARY_DIR}
)
endif()
endif()
endif()

Expand Down
Loading