Skip to content

Commit

Permalink
Find the library instead of assuming a path.
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Nov 28, 2024
1 parent 2a1071b commit 66870b6
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion ci/test_wheel_cuvs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,13 @@ python -m pip install $(echo ./dist/cuvs*.whl)[test]
# arm tests sporadically run into https://bugzilla.redhat.com/show_bug.cgi?id=1722181.
# This is a workaround to ensure that OpenMP gets the TLS that it needs.
if [[ "$(arch)" == "aarch64" ]]; then
export LD_PRELOAD=/opt/conda/envs/test/lib/libgomp.so.1
for dir in $(ld --verbose | grep SEARCH_DIR | tr -s ' ;' \\012 | grep -o '=.*"' | sed 's/[="]//g'); do
lib="${dir}/libgomp.so.1"
if [[ -f "${lib}" ]]; then
export LD_PRELOAD="${lib}"
break
fi
done
fi

python -m pytest ./python/cuvs/cuvs/test

0 comments on commit 66870b6

Please sign in to comment.