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
14 lines
256 B
C
4 years ago
|
#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
|