Skip to content

Commit

Permalink
Merge branch 'master' of github.com:clementfarabet/buffer
Browse files Browse the repository at this point in the history
  • Loading branch information
clementfarabet committed Jul 11, 2014
2 parents 1950530 + 52088a3 commit 5011d51
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions init.lua
Expand Up @@ -408,37 +408,37 @@ if torch then
local size = self.length / 4
local raw = ffi.cast('float *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.FloatStorage(size, pointer)
end
function Buffer:toLongStorage()
local size = self.length / 8
local raw = ffi.cast('long *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.LongStorage(size, pointer)
end
function Buffer:toIntStorage()
local size = self.length / 4
local raw = ffi.cast('int *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.IntStorage(size, pointer)
end
function Buffer:toShortStorage()
local size = self.length / 2
local raw = ffi.cast('short *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.ShortStorage(size, pointer)
end
function Buffer:toCharStorage()
local size = self.length
local raw = ffi.cast('char *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.CharStorage(size, pointer)
end
function Buffer:toByteStorage()
local size = self.length
local raw = ffi.cast('unsigned char *', self.ctype)
local pointer = tonumber(ffi.cast('long', raw))
return torch.DoubleStorage(size, pointer)
return torch.ByteStorage(size, pointer)
end

-- Auto gen tensors from storages:
Expand Down

0 comments on commit 5011d51

Please sign in to comment.