Incorrect value returned.

torch · Jun 5, 2014 · 9ac0c06 · 9ac0c06
1 parent 26a89bc
commit 9ac0c06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MultiMarginCriterion.cu b/MultiMarginCriterion.cu
@@ -141,8 +141,8 @@ static int cunn_MultiMarginCriterion_updateOutput(lua_State *L)
 static int cunn_MultiMarginCriterion_updateGradInput(lua_State *L)
 {
   THCudaTensor *input = (THCudaTensor*)luaT_checkudata(L, 2, "torch.CudaTensor");
-  THCudaTensor *gradInput = (THCudaTensor*)luaT_getfieldcheckudata(L, 1, "gradInput", "torch.CudaTensor");
   int sizeaverage = luaT_getfieldcheckboolean(L, 1, "sizeAverage");
+  THCudaTensor *gradInput = (THCudaTensor*)luaT_getfieldcheckudata(L, 1, "gradInput", "torch.CudaTensor");
 
   THCudaTensor_resizeAs(gradInput, input);