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.
13 lines
208 B
C
13 lines
208 B
C
4 years ago
|
#ifndef _SEARCHSORTED_CUDA_KERNEL
|
||
|
#define _SEARCHSORTED_CUDA_KERNEL
|
||
|
|
||
|
#include <torch/extension.h>
|
||
|
|
||
|
void searchsorted_cuda(
|
||
|
at::Tensor a,
|
||
|
at::Tensor v,
|
||
|
at::Tensor res,
|
||
|
bool side_left);
|
||
|
|
||
|
#endif
|