diff options
author | Dominik Grewe <dominikg@google.com> | 2015-04-10 17:18:46 +0300 |
---|---|---|
committer | Dominik Grewe <dominikg@google.com> | 2015-05-19 13:40:55 +0300 |
commit | 7800d7d8e3f3751837c2d08cf0dc9799d0dfe37b (patch) | |
tree | f5ebaf7e7a9bbca2a5a33180c7c4b11ac1cd0af5 /FFI.lua | |
parent | 2bffab89109313e2f5cc57aaf63866b9fe681f21 (diff) |
Add CudaHostAllocator
* A new allocator that uses cudaMallocHost.
* cutorch.createCudaHostTensor(...) to create FloatTensor allocated with
CudaHostAllocator.
Diffstat (limited to 'FFI.lua')
-rw-r--r-- | FFI.lua | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -14,6 +14,7 @@ typedef struct THCState int numDevices; int numUserStreams; int currentPerDeviceStream; + struct THAllocator* cudaHostAllocator; } THCState; cudaStream_t THCState_getCurrentStream(THCState *state); |