allow non-contiguous gradOutput in LookupTable

1 files changed, 7 insertions, 0 deletions

diff --git a/LookupTable.lua b/LookupTable.lua
index 95bf0fc..88eabab 100644
--- a/LookupTable.lua
+++ b/LookupTable.lua
@@ -70,6 +70,12 @@ function LookupTable:accGradParameters(input, gradOutput, scale)
       error("input must be a vector or matrix")
    end
 
+   if not gradOutput:isContiguous() then
+       self._gradOutput = self._gradOutput or gradOutput.new()
+       self._gradOutput:resizeAs(gradOutput):copy(gradOutput)
+       gradOutput = self._gradOutput
+   end
+
    self.gradWeight.THNN.LookupTable_accGradParameters(
       input:cdata(),
       gradOutput:cdata(),
@@ -101,6 +107,7 @@ function LookupTable:type(type, tensorCache)
 end
 
 function LookupTable:clearState()
+   self._gradOutput = nil
    return self
 end