pwrliang commented on PR #766: URL: https://github.com/apache/sedona-db/pull/766#issuecomment-4260274501
We’ll eventually need a CI server with a GPU if we want to get serious about GPU support @jiayuasu @paleolimbot . Test results from my server are included below: Note that this PR depends on https://github.com/apache/sedona-db/pull/766, or the GPU join might not trigger correctly. ``` (sedona) ubuntu@ip-172-31-42-84:/mnt/data/sedona-db/python/sedonadb$ python -m pytest tests/test_sjoin_gpu.py -vv -s ================================================================================================ test session starts ================================================================================================= platform linux -- Python 3.11.14, pytest-9.0.2, pluggy-1.6.0 -- /home/ubuntu/miniconda3/envs/sedona/bin/python cachedir: .pytest_cache rootdir: /mnt/data/sedona-db/python/sedonadb configfile: pyproject.toml plugins: anyio-4.12.1 collected 28 items tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-INNER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-INNER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-INNER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-LEFT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-RIGHT OUTER JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_reordering_can_be_disabled_e2e_gpu PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-LEFT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-RIGHT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-LEFT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-LEFT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-RIGHT SEMI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_join_semi_anti_gpu[ST_Contains(sjoin_polygon.geometry, sjoin_point.geometry)-RIGHT ANTI JOIN] PASSED tests/test_sjoin_gpu.py::test_spatial_mark_join_via_correlated_exists_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-point] PASSED tests/test_sjoin_gpu.py::test_spatial_mark_join_via_correlated_exists_gpu[ST_Intersects(sjoin_point.geometry, sjoin_polygon.geometry)-polygon] PASSED tests/test_sjoin_gpu.py::test_spatial_mark_join_via_correlated_exists_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-point] PASSED tests/test_sjoin_gpu.py::test_spatial_mark_join_via_correlated_exists_gpu[ST_Within(sjoin_point.geometry, sjoin_polygon.geometry)-polygon] PASSED tests/test_sjoin_gpu.py::test_query_window_in_subquery_gpu PASSED tests/test_sjoin_gpu.py::test_spatial_join_with_pandas_metadata_gpu PASSED ``` -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
