Skip to content

Commit

Permalink
Fix test cmake and add file
Browse files Browse the repository at this point in the history
  • Loading branch information
ggeorgakoudis committed Nov 4, 2023
1 parent f5f6940 commit 769d73c
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ ADDTEST(ams_inference_double torch_model.cpp AMSInferDouble ${CMAKE_CURRENT_SOUR
ADDTEST(ams_inference_single torch_model.cpp AMSInferSingle ${CMAKE_CURRENT_SOURCE_DIR}/debug_model.pt "single")
endif()

add_test(NAME AMSInferSingleDeltaUQ::HOST COMMAND ams_example --precision single --uqtype deltauq-mean -db ./db -S /p/vast1/ggeorgak/projects/ams/deltauq/tuple-single.torchscript -e 100)
add_test(NAME AMSInferSingleDeltaUQ::HOST COMMAND ams_example --precision single --uqtype deltauq-mean -db ./db -S ${CMAKE_CURRENT_SOURCE_DIR}/tuple-single.torchscript -e 100)
if(WITH_FAISS)
ADDTEST(ams_hdcache_mean_double test_hdcache.cpp AMSHDCacheMeanPolicyDouble ${CMAKE_CURRENT_SOURCE_DIR}/faiss_debug.pt "double" 1 10 4.0 4 5)
# The max case fails on DEVICE. We should be aware abou this when adding support for CI for GPUs
Expand Down
Binary file added tests/tuple-single.torchscript
Binary file not shown.

0 comments on commit 769d73c

Please sign in to comment.