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

3
unsupported/test/cxx11_tensor_contract_cuda.cpp
View File

@@ -52,8 +52,7 @@ static void test_cuda_contraction(int m_size, int k_size, int n_size)
cudaMemcpy(d_t_left, t_left.data(), t_left_bytes, cudaMemcpyHostToDevice);
cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice);
cudaStream_t stream;
assert(cudaStreamCreate(&stream) == cudaSuccess);
Eigen::CudaStreamDevice stream;
Eigen::GpuDevice gpu_device(&stream);
Eigen::TensorMap<Eigen::Tensor<float, 2, DataLayout> >