Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/cutorch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDominik Grewe <dominikg@google.com>2015-04-10 17:18:46 +0300
committerDominik Grewe <dominikg@google.com>2015-05-19 13:40:55 +0300
commit7800d7d8e3f3751837c2d08cf0dc9799d0dfe37b (patch)
treef5ebaf7e7a9bbca2a5a33180c7c4b11ac1cd0af5 /FFI.lua
parent2bffab89109313e2f5cc57aaf63866b9fe681f21 (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.lua1
1 files changed, 1 insertions, 0 deletions
diff --git a/FFI.lua b/FFI.lua
index 6aaec81..1b815cb 100644
--- a/FFI.lua
+++ b/FFI.lua
@@ -14,6 +14,7 @@ typedef struct THCState
int numDevices;
int numUserStreams;
int currentPerDeviceStream;
+ struct THAllocator* cudaHostAllocator;
} THCState;
cudaStream_t THCState_getCurrentStream(THCState *state);