File size: 1,089 Bytes
fadb92b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#ifdef __cplusplus
extern "C" {
  #endif

  // CUDA forward declarations
  std::vector<at::Tensor> forward_rasterize_cuda(at::Tensor vertices,
                                                 at::Tensor rasterized,
                                                 at::Tensor contribution_map,
                                                 int width,
                                                 int height,
                                                 float inv_smoothness,
                                                 int mode);

  at::Tensor backward_rasterize_cuda(at::Tensor vertices,
                                     at::Tensor rasterized,
                                     at::Tensor contribution_map,
                                     at::Tensor grad_output,
                                     at::Tensor grad_vertices,
                                     int width,
                                     int height,
                                     float inv_smoothness,
                                     int mode);

  #ifdef __cplusplus
}
#endif