NVIDIAGameWorks / kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).
Other
1.45k stars 132 forks source link

BaseAS interface #99

Closed orperel closed 1 year ago

orperel commented 1 year ago

This pull request adds the BaseAS abstract, to unify the upcoming acceleration structures under a single interface.

Notably raymarch / raytrace / query now return dataclasses of results. This is desired because:

  1. It allows to easily handle and pass around the various return values from these functions
  2. In case some raymarch ops return other output formats in the future, we could more easily enforce it by querying the dataclass return type