423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-25 08:40:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make PyTorch3D C++17 incompatible again :(

Browse Source 
Summary: D38919607 (c4545a7cbc) and D38858887 (06cbba2628) were premature, turns out CUDA 10.2 doesn't support C++17.

Reviewed By: bottler

Differential Revision: D39156205

fbshipit-source-id: 5e2e84cc4a57d1113a915166631651d438540d56
This commit is contained in:
Krzysztof Chalupka
2022-08-31 02:38:31 -07:00
committed by Facebook GitHub Bot
parent 1530a66469
commit 95771985b7
3 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pytorch3d/csrc/iou_box3d/iou_box3d.cu
View File

@@ -29,10 +29,10 @@ __global__ void IoUBox3DKernel(
const size_t tid = blockIdx.x * blockDim.x + threadIdx.x;
const size_t stride = gridDim.x * blockDim.x;
std::array<FaceVerts, NUM_TRIS> box1_tris{};
std::array<FaceVerts, NUM_TRIS> box2_tris{};
std::array<FaceVerts, NUM_PLANES> box1_planes{};
std::array<FaceVerts, NUM_PLANES> box2_planes{};
FaceVerts box1_tris[NUM_TRIS];
FaceVerts box2_tris[NUM_TRIS];
FaceVerts box1_planes[NUM_PLANES];
FaceVerts box2_planes[NUM_PLANES];
for (size_t i = tid; i < N * M; i += stride) {
const size_t n = i / M; // box1 index