sigmoid_focal_loss_forward_cuda_kernel not implemented for "Bfloat16"

[Fightinging1997]

Prerequisite

• I have searched Issues and Discussions but cannot get the expected help.
• 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]

Have you been able to solve it?