Incompatibility with bfloat16

[PabloVD]

trafficstars

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]

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

[PabloVD]

Great, thanks for the response!

[github-actions[bot]]

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?