CUDA error: no kernel image is available at polygon_b_inter_union_cuda

[UeFrog]

Hi, thanks for you great work. I managed to train my custom dataset,, but the inclusion of the function 'polygon_b_inter_union_cuda' is problematic.
The error thrown is:
File "/usr/local/etc/rotate-yolov5/utils/general.py", line 890, in polygon_bbox_iou
union += eps
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA is installed and available. If I use the CPU-version 'polygon_b_inter_union_cpu' everything works flawlessly, but the training is very slow. Do you know what the problem might be?
I use python 3.7.10, torch 1.10.0, cuda 11.3 on ubuntu 18.04