vhad/torch-mesh-isect
1
0
Fork 0
mirror of https://github.com/gosticks/torch-mesh-isect.git synced 2025-10-16 11:55:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: cuda update

Browse Source
This commit is contained in:
Wlad 2021-02-17 12:25:20 +01:00
parent d5201cc53b
commit cd0dcfe1e8
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/bvh.cpp
View File

@ -19,6 +19,10 @@
#include <vector>
#ifndef AT_CHECK
#define AT_CHECK TORCH_CHECK
#endif
void bvh_cuda_forward(at::Tensor triangles, at::Tensor *collision_tensor_ptr,
int max_collisions = 16);