mirror of
				https://github.com/facebookresearch/pytorch3d.git
				synced 2025-11-04 18:02:14 +08:00 
			
		
		
		
	remove IntWrapper (#1964)
Summary: I could not access https://github.com/NVlabs/cub/issues/172 to understand whether IntWrapper was still necessary but the comment is from 5 years ago and causes problems for the ROCm build. Pull Request resolved: https://github.com/facebookresearch/pytorch3d/pull/1964 Reviewed By: MichaelRamamonjisoa Differential Revision: D71937895 Pulled By: bottler fbshipit-source-id: 5e0351e1bd8599b670436cd3464796eca33156f6
This commit is contained in:
		
							parent
							
								
									3987612062
								
							
						
					
					
						commit
						0a59450f0e
					
				@ -70,11 +70,6 @@ struct CamGradInfo {
 | 
				
			|||||||
  float3 pixel_dir_y;
 | 
					  float3 pixel_dir_y;
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// TODO: remove once https://github.com/NVlabs/cub/issues/172 is resolved.
 | 
					 | 
				
			||||||
struct IntWrapper {
 | 
					 | 
				
			||||||
  int val;
 | 
					 | 
				
			||||||
};
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
} // namespace pulsar
 | 
					} // namespace pulsar
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#endif
 | 
					#endif
 | 
				
			||||||
 | 
				
			|||||||
@ -149,11 +149,6 @@ IHD CamGradInfo operator*(const CamGradInfo& a, const float& b) {
 | 
				
			|||||||
  return res;
 | 
					  return res;
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
IHD IntWrapper operator+(const IntWrapper& a, const IntWrapper& b) {
 | 
					 | 
				
			||||||
  IntWrapper res;
 | 
					 | 
				
			||||||
  res.val = a.val + b.val;
 | 
					 | 
				
			||||||
  return res;
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
} // namespace pulsar
 | 
					} // namespace pulsar
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#endif
 | 
					#endif
 | 
				
			||||||
 | 
				
			|||||||
@ -155,8 +155,8 @@ void backward(
 | 
				
			|||||||
        stream);
 | 
					        stream);
 | 
				
			||||||
    CHECKLAUNCH();
 | 
					    CHECKLAUNCH();
 | 
				
			||||||
    SUM_WS(
 | 
					    SUM_WS(
 | 
				
			||||||
        (IntWrapper*)(self->ids_sorted_d),
 | 
					        self->ids_sorted_d,
 | 
				
			||||||
        (IntWrapper*)(self->n_grad_contributions_d),
 | 
					        self->n_grad_contributions_d,
 | 
				
			||||||
        static_cast<int>(num_balls),
 | 
					        static_cast<int>(num_balls),
 | 
				
			||||||
        self->workspace_d,
 | 
					        self->workspace_d,
 | 
				
			||||||
        self->workspace_size,
 | 
					        self->workspace_size,
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user