diff options
author | Determinant <ted.sybil@gmail.com> | 2016-03-02 14:59:59 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-03-02 14:59:59 +0800 |
commit | 75108e2e000f4382129d453c0fa9073b14d32f97 (patch) | |
tree | 9b952b8a39affdc7725354ec41940ce5af9d4629 /nerv/matrix | |
parent | 42608086c666e23ceac0e9046836a8fefe00e5b2 (diff) |
add options to main entry `nerv`; fix a bug in handling `cudaSetDevice` error
Diffstat (limited to 'nerv/matrix')
-rw-r--r-- | nerv/matrix/cumatrix.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/matrix/cumatrix.c b/nerv/matrix/cumatrix.c index f6a4ed5..7d10895 100644 --- a/nerv/matrix/cumatrix.c +++ b/nerv/matrix/cumatrix.c @@ -9,7 +9,7 @@ const char *nerv_cuda_context_tname = "nerv.CuContext"; int nerv_cuda_context_lua_select_gpu(lua_State *L) { Status status; nerv_cuda_context_select_gpu(luaT_checkudata(L, 1, nerv_cuda_context_tname), - luaL_checkinteger(L, 1), &status); + luaL_checkinteger(L, 2), &status); NERV_LUA_CHECK_STATUS(L, status); return 0; } |