devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated the cuda tests to use the new GpuDevice constructor

Browse Source
This commit is contained in:
Benoit Steiner
2015-07-15 12:39:26 -07:00
parent e892524efe
commit 8315e025e1
3 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
unsupported/test/cxx11_tensor_device.cpp
View File

@@ -77,14 +77,11 @@ struct GPUContext {
assert(cudaMalloc((void**)(&kernel_3d_), 8*sizeof(float)) == cudaSuccess);
float kernel_3d_val[] = {3.14f, -1.0f, 2.7f, -0.3f, 0.2f, -0.7f, 7.0f, -0.5f};
assert(cudaMemcpy(kernel_3d_, kernel_3d_val, 8*sizeof(float), cudaMemcpyHostToDevice) == cudaSuccess);
assert(cudaStreamCreate(&stream_) == cudaSuccess);
}
~GPUContext() {
assert(cudaFree(kernel_1d_) == cudaSuccess);
assert(cudaFree(kernel_2d_) == cudaSuccess);
assert(cudaFree(kernel_3d_) == cudaSuccess);
assert(cudaStreamDestroy(stream_) == cudaSuccess);
}
const Eigen::GpuDevice& device() const { return gpu_device_; }
@@ -105,7 +102,7 @@ struct GPUContext {
float* kernel_2d_;
float* kernel_3d_;
cudaStream_t stream_;
Eigen::CudaStreamDevice stream_;
Eigen::GpuDevice gpu_device_;
};