search

masadcv / FastGeodis

Fast Implementation of Generalised Geodesic Distance Transform for CPU (OpenMP) and GPU (CUDA)
https://fastgeodis.readthedocs.io
BSD 3-Clause "New" or "Revised" License
90 stars 14 forks source link

Fix cuda target on multi-gpu host #44

Closed masadcv closed 1 year ago

masadcv commented 1 year ago

Attempting to fix cuda device error on multi-gpu device, see #43 for more details.

The solution is to utilize CUDA Guard link.

Tested locally on 2 GPU machine