binding.cpp File Reference

#include "ray.h"
#include <torch/extension.h>

Include dependency graph for binding.cpp:

Go to the source code of this file.

Namespaces
namespace	hmesh

Functions
void	hmesh::initOptix ()
void	hmesh::createOptixContext ()
void	hmesh::createOptixModule ()
torch::Tensor	hmesh::intersectsAny (OptixAccelStructureWrapperCPP as, const torch::Tensor &origins, const torch::Tensor &dirs)
torch::Tensor	hmesh::intersectsFirst (OptixAccelStructureWrapperCPP as, const torch::Tensor &origins, const torch::Tensor &dirs)
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor >	hmesh::intersectsClosest (OptixAccelStructureWrapperCPP as, torch::Tensor origins, torch::Tensor directions)
	Find if ray hits any triangle and return ray index, triangle index, hit location and uv.
torch::Tensor	hmesh::intersectsCount (OptixAccelStructureWrapperCPP as, torch::Tensor origins, torch::Tensor directions)
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	hmesh::intersectsLocation (OptixAccelStructureWrapperCPP as, torch::Tensor origins, torch::Tensor directions)
void	hmesh::createPipelines ()
void	hmesh::buildSBT ()
	PYBIND11_MODULE (TORCH_EXTENSION_NAME, m)

Function Documentation

PYBIND11_MODULE()

PYBIND11_MODULE	(	TORCH_EXTENSION_NAME	,
		m
	)

Definition at line 31 of file binding.cpp.