cpu.c 377 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
#include <TH/TH.h>

rusty1s's avatar
rename  
rusty1s committed
3
#include "THTensorDimApply4.h"
rusty1s's avatar
rusty1s committed
4

rusty1s's avatar
rusty1s committed
5
#define scatter_(NAME) TH_CONCAT_4(scatter_, NAME, _, Real)
rusty1s's avatar
rusty1s committed
6
#define index_backward TH_CONCAT_2(index_backward_, Real)
rusty1s's avatar
rusty1s committed
7

rusty1s's avatar
rusty1s committed
8
inline void assertIndexInBoundaries(int idx, int size, int64_t *free) {
rusty1s's avatar
rusty1s committed
9
10
11
  if (idx < 0 || idx >= size) { THFree(free); THError("Invalid index"); }
}

rusty1s's avatar
rusty1s committed
12
#include "generic/cpu.c"
rusty1s's avatar
rusty1s committed
13
#include "THGenerateAllTypes.h"