Skip to content

Small Updation for Cuda >= 11.0 and Pytorch >= 1.0 #31

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
prajwalsingh wants to merge 2 commits into
base: master
Choose a base branch
from
+7 −2
Open

Small Updation for Cuda >= 11.0 and Pytorch >= 1.0 #31

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
21f5fab
Update bvh.cpp
prajwalsingh Oct 5, 2022
4bf7fbe
Update README.md
prajwalsingh Oct 5, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -136,3 +136,8 @@ The code of this repository was implemented by [Vassilis Choutas](vassilis.chout
For questions, please contact [smplx@tue.mpg.de](smplx@tue.mpg.de).

For commercial licensing (and all related questions for business applications), please contact [ps-licensing@tue.mpg.de](ps-licensing@tue.mpg.de). Please note that the method for this component has been [patented by NVidia](https://patents.google.com/patent/US9396512B2/en) and a license needs to be obtained also by them.

## Acknowledgement
* [error: ‘AT_CHECK’ was not declared in this scope](https://github.com/open-mmlab/OpenPCDet/issues/284#issuecomment-874171209): Fix for latest Pytorch versions

* [error: Missing helper_math.h](https://github.com/OpenKinect/libfreenect2/issues/777#issuecomment-997941415): Only second and third step.
4 changes: 2 additions & 2 deletions src/bvh.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,8 @@
void bvh_cuda_forward(at::Tensor triangles, at::Tensor* collision_tensor_ptr,
int max_collisions = 16);

#define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x " must be a CUDA tensor")
#define CHECK_CONTIGUOUS(x) AT_CHECK(x.is_contiguous(), #x " must be contiguous")
#define CHECK_CUDA(x) TORCH_CHECK(x.type().is_cuda(), #x " must be a CUDA tensor")
#define CHECK_CONTIGUOUS(x) TORCH_CHECK(x.is_contiguous(), #x " must be contiguous")
#define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x)

at::Tensor bvh_forward(at::Tensor triangles, int max_collisions = 16) {
Expand Down