Add THHalfTensor support to cutorch (#655)

* Add THHalfTensor support to cutorch.

1 files changed, 7 insertions, 0 deletions

diff --git a/Tensor.c b/Tensor.c
index 3adeddd..560d224 100644
--- a/Tensor.c
+++ b/Tensor.c
@@ -8,6 +8,13 @@
 #define torch_Tensor_(NAME) TH_CONCAT_4(torch_,CReal,Tensor_,NAME)
 #define torch_Tensor TH_CONCAT_STRING_3(torch.,CReal,Tensor)
 #define cutorch_Tensor_(NAME) TH_CONCAT_4(cutorch_,CReal,Tensor_,NAME)
+#define cutorch_TensorCopy_(NAME) TH_CONCAT_4(cutorch_,Real,TensorCopy_,NAME)
 
+// generate the torch types -- we could also do this via THGenerateAllTypes,
+// but this allows us to be self contained.
+#define FORCE_TH_HALF
+#include "generic/CTensorCopy.c"
+#include "THCGenerateAllTypes.h"
+#undef FORCE_TH_HALF
 #include "generic/CTensor.c"
 #include "THCGenerateAllTypes.h"