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

use less thrust, maybe help Windows

Browse Source 
Summary: I think we include more thrust than needed, and maybe removing it will help things like  https://github.com/facebookresearch/pytorch3d/issues/1610 with DebugSyncStream errors on Windows.

Reviewed By: shapovalov

Differential Revision: D48949888

fbshipit-source-id: add889c0acf730a039dc9ffd6bbcc24ded20ef27
This commit is contained in:
Jeremy Reizenstein
2023-09-05 04:40:24 -07:00
committed by Facebook GitHub Bot
parent a3d99cab6b
commit 6f2212da46
3 changed files with 18 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,8 +12,6 @@
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <thrust/device_vector.h>
#include <thrust/tuple.h>
#include "iou_box3d/iou_utils.cuh"
// Parallelize over N*M computations which can each be done