triro 1.3.0
A Python Ray-Mesh Intersector in OptiX
|
evrything per-instance More...
#include "ray.h"
#include "ATen/core/TensorBody.h"
#include "ATen/ops/where.h"
#include "CUDABuffer.h"
#include "LaunchParams.h"
#include "base.h"
#include "c10/core/Layout.h"
#include "c10/core/ScalarType.h"
#include "c10/core/TensorOptions.h"
#include "c10/util/ArrayRef.h"
#include "optix8.h"
#include "optix_host.h"
#include "optix_types.h"
#include "sbtdef.h"
#include "type.h"
#include <limits>
Go to the source code of this file.
Namespaces | |
namespace | hmesh |
Functions | |
template<typename... Ts> | |
bool | hmesh::tensorInputCheck (Ts... ts) |
std::vector< int64_t > | hmesh::removeLastDim (const c10::IntArrayRef dims) |
size_t | hmesh::prod (const std::vector< int64_t > &dims) |
std::vector< int64_t > | hmesh::changeLastDim (const c10::IntArrayRef dims, size_t value) |
template<typename T > | |
T * | hmesh::data_ptr (const torch::Tensor &t) |
template<typename T > | |
void | hmesh::fillArray (T *dst, c10::ArrayRef< T > src, T defaultValue) |
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) |
evrything per-instance
Definition in file ray.cpp.