OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

corrected bug in method type-base dispatcher

Browse Source
This commit is contained in:
Ronan Collobert 2012-02-10 13:43:49 +01:00
parent bc15365ca5
commit 9ca97814d3
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
TensorMath.lua
View File

@ -170,16 +170,19 @@ static int torch_NAME(lua_State *L)
int narg = lua_gettop(L);
const void *id;
if(narg < 1 || !(id = torch_istensorid(L, luaT_id(L, 1)))) /* first argument is tensor? */
if(narg >= 1 && (id = torch_istensorid(L, luaT_id(L, 1)))) /* first argument is tensor? */
{
if(narg < 2 || !(id = torch_istensorid(L, luaT_id(L, 2)))) /* second? */
{
if(lua_isstring(L, -1) && (id = torch_istensorid(L, luaT_typename2id(L, lua_tostring(L, -1))))) /* do we have a valid string then? */
lua_remove(L, -2);
else if(!(id = torch_istensorid(L, torch_getdefaulttensorid())))
luaL_error(L, "internal error: the default tensor type does not seem to be an actual tensor");
}
}
else if(narg >= 2 && (id = torch_istensorid(L, luaT_id(L, 2)))) /* second? */
{
}
else if(narg >= 1 && lua_isstring(L, -1)
&& (id = torch_istensorid(L, luaT_typename2id(L, lua_tostring(L, -1))))) /* do we have a valid string then? */
{
lua_remove(L, -2);
}
else if(!(id = torch_istensorid(L, torch_getdefaulttensorid())))
luaL_error(L, "internal error: the default tensor type does not seem to be an actual tensor");
lua_pushstring(L, "NAME");
lua_rawget(L, -2);