search

ThibaultGROUEIX / ChamferDistancePytorch

Chamfer Distance in Pytorch with f-score
MIT License
326 stars 43 forks source link

fp16 compatibility #22

Open kampelmuehler opened 2 years ago

kampelmuehler commented 2 years ago

Thanks for the great repo! Can you point me to what needs to be done to enable fp16 compatibility? I suppose it will need some rewriting in the CUDA kernels. Is it just changing the types in the function declarations or more work that needs to go into it?

ThibaultGROUEIX commented 1 year ago

Hi @kampelmuehler I think it is just changing the types, that's a great idea. You're welcome to make a PR for this. Cheers, Thibault

ThibaultGROUEIX commented 1 year ago

Though now looking at the date of your issue, you may not be interested in this feature anymore

kampelmuehler commented 1 year ago

Thanks, you're right - my interest has dwindled meanwhile