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

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

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

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

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