jstmn / Jrl

Jeremy's Robotics Library
MIT License
37 stars 3 forks source link

nvidia warp for geodesic_distance_between_quaternions() #10

Closed jstmn closed 9 months ago

jstmn commented 9 months ago

This pr adds a warp implementation for geodesic_distance_between_quaternions(). An empirical runtime test was performed to determine which implementation is faster - the original pytorch version or this newly added warp version.

The results indicate that the original pytorch version is faster for all batch sizes.

geodesic_distance() runtime comparison - pytorch vs warp

Regardless, the code is going to added as a reference. Also, the pr includes other minor refactors and cleanups and I was too lazy to parse them out,