kernel.cu 652 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
#include <THC/THC.h>

3
#include "kernel.h"
rusty1s's avatar
rusty1s committed
4
#include "common.cuh"
rusty1s's avatar
rusty1s committed
5

rusty1s's avatar
rusty1s committed
6
7
#define scatter_(NAME) TH_CONCAT_4(scatter_, NAME, _kernel_, Real)
#define index_backward TH_CONCAT_2(index_backward_kernel_, Real)
rusty1s's avatar
rusty1s committed
8
#define check TH_CONCAT_2(check_kernel_, Real)
9

rusty1s's avatar
rusty1s committed
10
11
12
13
14
15
16
17
18
19
20
#define thc_(NAME) TH_CONCAT_4(thc_, NAME, _, Real)

#include "generic/common.cu"
#include "THCGenerateAllTypes.h"

template <typename Real, int Dims>
__global__ void maxKernel(TensorInfo<Real> output, TensorInfo<int64_t> index, TensorInfo<Real> input, TensorInfo<int64_t> arg_output, const int dim, const int n) {
  KERNEL_LOOP(i, n) {

  }
}
rusty1s's avatar
max dim  
rusty1s committed
21

rusty1s's avatar
rusty1s committed
22
23
#include "generic/kernel.cu"
#include "THCGenerateAllTypes.h"