|
| | ops.get_module () |
| |
| | ops.init_optix () |
| |
| | ops.create_optix_context () |
| |
| | ops.create_optix_module () |
| |
| | ops.create_optix_pipelines () |
| |
| | ops.build_sbts () |
| |
| Bool[torch.Tensor, "*b"] | ops.intersects_any (OptixAccelStructureWrapper accel_structure, Float32[torch.Tensor, "*b 3"] origins, Float32[torch.Tensor, "*b 3"] dirs) |
| |
| Int32[torch.Tensor, "*b"] | ops.intersects_first (OptixAccelStructureWrapper accel_structure, Float32[torch.Tensor, "*b 3"] origins, Float32[torch.Tensor, "*b 3"] dirs) |
| |
| Tuple[ Bool[torch.Tensor, "*b"], # hit Bool[torch.Tensor, "*b"], # front Int32[torch.Tensor, "*b"], # triangle index Float32[torch.Tensor, "*b 3"], # intersect location Float32[torch.Tensor, "*b 2"], # uv] | ops.intersects_closest (OptixAccelStructureWrapper accel_structure, Float32[torch.Tensor, "*b 3"] origins, Float32[torch.Tensor, "*b 3"] dirs) |
| |
| Int32[torch.Tensor, "*b"] | ops.intersects_count (OptixAccelStructureWrapper accel_structure, Float32[torch.Tensor, "*b 3"] origins, Float32[torch.Tensor, "*b 3"] dirs) |
| |
| Tuple[ Float32[torch.Tensor, "h 3"], Int32[torch.Tensor, "h"], Int32[torch.Tensor, "h"]] | ops.intersects_location (OptixAccelStructureWrapper accel_structure, Float32[torch.Tensor, "*b 3"] origins, Float32[torch.Tensor, "*b 3"] dirs) |
| |