423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-03-11 23:06:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

nondeterminism warnings

Browse Source 
Summary: do like xformers.

Reviewed By: shapovalov

Differential Revision: D44541873

fbshipit-source-id: 2c23160591cd9026fcd4972998d1bc90adba1356
This commit is contained in:
Jeremy Reizenstein
2023-05-04 12:50:41 -07:00
committed by Facebook GitHub Bot
parent 3e3644e534
commit ef5f620263
7 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch3d/csrc/rasterize_meshes/rasterize_meshes.cu
View File

@@ -583,6 +583,9 @@ at::Tensor RasterizeMeshesBackwardCuda(
at::checkAllSameType(
c, {face_verts_t, grad_zbuf_t, grad_bary_t, grad_dists_t});
// This is nondeterministic because atomicAdd
at::globalContext().alertNotDeterministic("RasterizeMeshesBackwardCuda");
// Set the device for the kernel launch based on the device of the input
at::cuda::CUDAGuard device_guard(face_verts.device());
cudaStream_t stream = at::cuda::getCurrentCUDAStream();