diff options
Diffstat (limited to 'nerv/matrix/init.lua')
-rw-r--r-- | nerv/matrix/init.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/nerv/matrix/init.lua b/nerv/matrix/init.lua index 1091d7e..50bdb11 100644 --- a/nerv/matrix/init.lua +++ b/nerv/matrix/init.lua @@ -53,19 +53,19 @@ nerv.MMatrixInt.fmt = "%d " function nerv.CuMatrix:__add__(b) c = self:create() - c:add(self, b, 1.0, 1.0) + c:add(self, b, 1.0, 1.0, nerv.context) return c end function nerv.CuMatrix:__sub__(b) c = self:create() - c:add(self, b, 1.0, -1.0) + c:add(self, b, 1.0, -1.0, nerv.context) return c end function nerv.CuMatrix:__mul__(b) c = nerv.get_type(self.__typename)(self:nrow(), b:ncol()) - c:mul(self, b, 1.0, 0.0, 'N', 'N') + c:mul(self, b, 1.0, 0.0, 'N', 'N', nerv.context) return c end |