diff options
author | gchanan <gregchanan@gmail.com> | 2016-12-29 22:23:45 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2016-12-29 22:23:45 +0300 |
commit | 1ac06689dba1a4a672ed1fb3c3117000a46d7af5 (patch) | |
tree | 5edb7b33c8ef506cbe488fe5d83de6349e42aaeb /Storage.c | |
parent | 6b763fd55f9919ec2f1ccf58c962213e6fb755ea (diff) |
Add THHalfTensor support to cutorch (#655)
* Add THHalfTensor support to cutorch.
Diffstat (limited to 'Storage.c')
-rw-r--r-- | Storage.c | 7 |
1 files changed, 7 insertions, 0 deletions
@@ -6,7 +6,14 @@ #define torch_Storage_(NAME) TH_CONCAT_4(torch_,CReal,Storage_,NAME) #define torch_Storage TH_CONCAT_STRING_3(torch.,CReal,Storage) #define cutorch_Storage_(NAME) TH_CONCAT_4(cutorch_,CReal,Storage_,NAME) +#define cutorch_StorageCopy_(NAME) TH_CONCAT_4(cutorch_,Real,StorageCopy_,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/CStorageCopy.c" +#include "THCGenerateAllTypes.h" +#undef FORCE_TH_HALF #include "generic/CStorage.c" #include "THCGenerateAllTypes.h" |