sigmoid_focal_loss_forward_cuda_kernel not implemented for "Bfloat16" - Githubissues

open-mmlab / mmcv

OpenMMLab Computer Vision Foundation

https://mmcv.readthedocs.io/en/latest/

Apache License 2.0

5.73k stars 1.61k forks source link

sigmoid_focal_loss_forward_cuda_kernel not implemented for "Bfloat16" #3061

Open Fightinging1997 opened 3 months ago

Fightinging1997 commented 3 months ago

Prerequisite

[X] I have searched Issues and Discussions but cannot get the expected help.
[X] The bug has not been fixed in the latest version(https://github.com/open-mmlab/mmcv).

Environment

python3.8. cuda11.1

Reproduces the problem - code sample

File "/training/env/xxx/lib/python3.8/site-packages/mmcv/ops/focal_loss.py", line 59, in forward ext_module.sigmoid_focal_loss_forward( RuntimeError: "sigmoid_focal_loss_forward_cuda_kernel" not implemented for 'BFloat16'

Reproduces the problem - command or script

File "/training/env/xxx/lib/python3.8/site-packages/mmcv/ops/focal_loss.py", line 59, in forward ext_module.sigmoid_focal_loss_forward( RuntimeError: "sigmoid_focal_loss_forward_cuda_kernel" not implemented for 'BFloat16'

Reproduces the problem - error message

File "/training/env/xxx/lib/python3.8/site-packages/mmcv/ops/focal_loss.py", line 59, in forward ext_module.sigmoid_focal_loss_forward( RuntimeError: "sigmoid_focal_loss_forward_cuda_kernel" not implemented for 'BFloat16'

Additional information

No response

gerardmartin2 commented 2 months ago

Have you been able to solve it?