rusty1s/pytorch_cluster

Incompatibility with bfloat16

PabloVD opened this issue · 3 comments

PabloVD commented

Hi,

I tried to run torch_cluster.radius_graph in CUDA with bfloat16 data and I got the following error:

    return torch.ops.torch_cluster.radius(x, y, ptr_x, ptr_y, r,
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
                                          max_num_neighbors, num_workers)
RuntimeError: "_" not implemented for 'BFloat16'

Is torch_cluster compatible with torch.bfloat16 format?

Thanks in advance.

rusty1s commented

We just recently added support for this. You sadly need to install from master for now to get torch.bfloat16 support.

PabloVD commented

Great, thanks for the response!

github-actions commented

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?