Repository navigation
Expand file tree
/
Copy pathinit.c
More file actions
140 lines (119 loc) · 3.2 KB
/
Copy pathinit.c
File metadata and controls
140 lines (119 loc) · 3.2 KB
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
133
134
135
136
137
138
139
140
#include "luaT.h"
#include "THCGeneral.h"
#include "THCTensorRandom.h"
extern void cutorch_CudaStorage_init(lua_State* L);
extern void cutorch_CudaTensor_init(lua_State* L);
extern void cutorch_CudaTensorMath_init(lua_State* L);
static int cutorch_synchronize(lua_State *L)
{
cudaDeviceSynchronize();
return 0;
}
static int cutorch_getDevice(lua_State *L)
{
int device;
THCudaCheck(cudaGetDevice(&device));
device++;
lua_pushnumber(L, device);
return 1;
}
static int cutorch_deviceReset(lua_State *L)
{
THCudaCheck(cudaDeviceReset());
return 0;
}
static int cutorch_getDeviceCount(lua_State *L)
{
int ndevice;
THCudaCheck(cudaGetDeviceCount(&ndevice));
lua_pushnumber(L, ndevice);
return 1;
}
static int cutorch_setDevice(lua_State *L)
{
int device = (int)luaL_checknumber(L, 1)-1;
THCudaCheck(cudaSetDevice(device));
THCRandom_manualSeed(THCRandom_initialSeed());
return 0;
}
#define SET_DEVN_PROP(NAME) \
lua_pushnumber(L, prop.NAME); \
lua_setfield(L, -2, #NAME);
static int cutorch_getDeviceProperties(lua_State *L)
{
struct cudaDeviceProp prop;
int device = (int)luaL_checknumber(L, 1)-1;
THCudaCheck(cudaGetDeviceProperties(&prop, device));
lua_newtable(L);
SET_DEVN_PROP(canMapHostMemory);
SET_DEVN_PROP(clockRate);
SET_DEVN_PROP(computeMode);
SET_DEVN_PROP(deviceOverlap);
SET_DEVN_PROP(integrated);
SET_DEVN_PROP(kernelExecTimeoutEnabled);
SET_DEVN_PROP(major);
SET_DEVN_PROP(maxThreadsPerBlock);
SET_DEVN_PROP(memPitch);
SET_DEVN_PROP(minor);
SET_DEVN_PROP(multiProcessorCount);
SET_DEVN_PROP(regsPerBlock);
SET_DEVN_PROP(sharedMemPerBlock);
SET_DEVN_PROP(textureAlignment);
SET_DEVN_PROP(totalConstMem);
SET_DEVN_PROP(totalGlobalMem);
SET_DEVN_PROP(warpSize);
SET_DEVN_PROP(pciBusID);
SET_DEVN_PROP(pciDeviceID);
SET_DEVN_PROP(pciDomainID);
SET_DEVN_PROP(maxTexture1D);
SET_DEVN_PROP(maxTexture1DLinear);
size_t freeMem;
THCudaCheck(cudaMemGetInfo (&freeMem, NULL));
lua_pushnumber(L, freeMem);
lua_setfield(L, -2, "freeGlobalMem");
lua_pushstring(L, prop.name);
lua_setfield(L, -2, "name");
return 1;
}
static int cutorch_seed(lua_State *L)
{
unsigned long seed = THCRandom_seed();
lua_pushnumber(L, seed);
return 1;
}
static int cutorch_initialSeed(lua_State *L)
{
unsigned long seed = THCRandom_initialSeed();
lua_pushnumber(L, seed);
return 1;
}
static int cutorch_manualSeed(lua_State *L)
{
unsigned long seed = luaL_checknumber(L, 1);
THCRandom_manualSeed(seed);
return 0;
}
static const struct luaL_Reg cutorch_stuff__ [] = {
{"synchronize", cutorch_synchronize},
{"getDevice", cutorch_getDevice},
{"deviceReset", cutorch_deviceReset},
{"getDeviceCount", cutorch_getDeviceCount},
{"getDeviceProperties", cutorch_getDeviceProperties},
{"setDevice", cutorch_setDevice},
{"seed", cutorch_seed},
{"initialSeed", cutorch_initialSeed},
{"manualSeed", cutorch_manualSeed},
{NULL, NULL}
};
LUA_EXTERNC DLL_EXPORT int luaopen_libcutorch(lua_State *L);
int luaopen_libcutorch(lua_State *L)
{
lua_newtable(L);
luaL_register(L, NULL, cutorch_stuff__);
THCudaInit();
THCRandom_seed();
cutorch_CudaStorage_init(L);
cutorch_CudaTensor_init(L);
cutorch_CudaTensorMath_init(L);
return 1;
}