nerf-pytorch/searchsorted_cuda_wrapper.h at bcb670095eefaf798d1cdde4592f2658a6b31085 - nerf-pytorch - Gitea: Service Git d'Aurore

Suppression de la possibilité de cloner des dépôts privés en HTTPS ;
Réinstallation de drone ou changement d'outil de CI ;
Possiblement migration de Gitea à Forgejo.

You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

14 lines

256 B

C

Raw Blame History

 #ifndef _SEARCHSORTED_CUDA_WRAPPER
 #define _SEARCHSORTED_CUDA_WRAPPER
 #include <torch/extension.h>
 #include "searchsorted_cuda_kernel.h"
 void searchsorted_cuda_wrapper(
     at::Tensor a,
     at::Tensor v,
     at::Tensor res,
     bool side_left);
 #endif