1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/CStorage.c"
#else
/* everything is as the generic Storage.c, except few things (see below) */
#define THFile_readRealRaw(file, data, size) \
{ \
real *fdata = (real*)THAlloc(sizeof(real)*size); \
TH_CONCAT_3(THFile_read,Real,Raw)(file, fdata, size); \
THCudaCheck(cudaMemcpy(data, fdata, size * sizeof(real), cudaMemcpyHostToDevice)); \
THFree(fdata); \
}
#define THFile_writeRealRaw(file, data, size) \
{ \
real *fdata = (real*)THAlloc(sizeof(real)*size); \
THCudaCheck(cudaMemcpy(fdata, data, size * sizeof(real), cudaMemcpyDeviceToHost)); \
TH_CONCAT_3(THFile_write,Real,Raw)(file, fdata, size); \
THFree(fdata); \
}
#define TH_GENERIC_FILE "generic/Storage.c"
#include "generic/Storage.c"
#undef TH_GENERIC_FILE
#undef THFile_readRealRaw
#undef THFile_writeRealRaw
/* now we overwrite some methods specific to CudaStorage */
static int cutorch_Storage_(copy)(lua_State *L)
{
THCState *state = cutorch_getstate(L);
THCStorage *storage = luaT_checkudata(L, 1, torch_Storage);
void *src;
if( (src = luaT_toudata(L, 2, "torch.CudaByteStorage")) )
THCStorage_(copyCudaByte)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaCharStorage")) )
THCStorage_(copyCudaChar)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaShortStorage")) )
THCStorage_(copyCudaShort)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaIntStorage")) )
THCStorage_(copyCudaInt)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaLongStorage")) )
THCStorage_(copyCudaLong)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaStorage")) )
THCStorage_(copyCudaFloat)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaDoubleStorage")) )
THCStorage_(copyCudaDouble)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.ByteStorage")) )
THCStorage_(copyByte)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CharStorage")) )
THCStorage_(copyChar)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.ShortStorage")) )
THCStorage_(copyShort)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.IntStorage")) )
THCStorage_(copyInt)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.LongStorage")) )
THCStorage_(copyLong)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.FloatStorage")) )
THCStorage_(copyFloat)(state, storage, src);
else if( (src = luaT_toudata(L, 2, "torch.DoubleStorage")) )
THCStorage_(copyDouble)(state, storage, src);
else
luaL_typerror(L, 2, "torch.*Storage");
lua_settop(L, 1);
return 1;
}
static int TH_CONCAT_3(cutorch_,Real,Storage_copy)(lua_State *L)
{
THStorage *storage = luaT_checkudata(L, 1, TH_CONCAT_STRING_3(torch.,Real,Storage));
void *src;
if( (src = luaT_toudata(L, 2, TH_CONCAT_STRING_3(torch.,Real,Storage) )))
THStorage_(copy)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.ByteStorage")) )
THStorage_(copyByte)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CharStorage")) )
THStorage_(copyChar)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.ShortStorage")) )
THStorage_(copyShort)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.IntStorage")) )
THStorage_(copyInt)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.LongStorage")) )
THStorage_(copyLong)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.FloatStorage")) )
THStorage_(copyFloat)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.DoubleStorage")) )
THStorage_(copyDouble)(storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaStorage")) )
THStorage_(copyCudaFloat)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaLongStorage")) )
THStorage_(copyCudaLong)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaByteStorage")) )
THStorage_(copyCudaByte)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaCharStorage")) )
THStorage_(copyCudaChar)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaShortStorage")) )
THStorage_(copyCudaShort)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaIntStorage")) )
THStorage_(copyCudaInt)(cutorch_getstate(L), storage, src);
else if( (src = luaT_toudata(L, 2, "torch.CudaDoubleStorage")) )
THStorage_(copyCudaDouble)(cutorch_getstate(L), storage, src);
else
luaL_typerror(L, 2, "torch.*Storage");
lua_settop(L, 1);
return 1;
}
void cutorch_Storage_(init)(lua_State* L)
{
/* the standard stuff */
torch_Storage_(init)(L);
// torch_Storage macro is defined in Storage.c produce the CudaTensor types
// so I have to construct the normal torch types by hand
luaT_pushmetatable(L, TH_CONCAT_STRING_3(torch.,Real,Storage));
lua_pushcfunction(L, TH_CONCAT_3(cutorch_,Real,Storage_copy));
lua_setfield(L, -2, "copy");
lua_pop(L, 1);
luaT_pushmetatable(L, torch_Storage);
lua_pushcfunction(L, cutorch_Storage_(copy));
lua_setfield(L, -2, "copy");
lua_pop(L, 1);
}
#endif
|
