GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 19:59:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix for float16 GPU unit test.

Browse Source
This commit is contained in:
Rohit Santhanam 2021-04-11 14:54:18 +00:00 committed by David Tellenbach
parent 1e1c8a735c
commit dfd6720d82
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/test/cxx11_tensor_of_float16_gpu.cu
View File

@ -461,7 +461,7 @@ void test_gpu_forced_evals() {
Tensor<float, 1> half_prec2(num_elem); Tensor<float, 1> half_prec2(num_elem);
Tensor<float, 1> full_prec(num_elem); Tensor<float, 1> full_prec(num_elem);
gpu_device.memcpyDeviceToHost(half_prec1.data(), d_res_half1, num_elem*sizeof(float)); gpu_device.memcpyDeviceToHost(half_prec1.data(), d_res_half1, num_elem*sizeof(float));
gpu_device.memcpyDeviceToHost(half_prec2.data(), d_res_half1, num_elem*sizeof(float)); gpu_device.memcpyDeviceToHost(half_prec2.data(), d_res_half2, num_elem*sizeof(float));
gpu_device.memcpyDeviceToHost(full_prec.data(), d_res_float, num_elem*sizeof(float)); gpu_device.memcpyDeviceToHost(full_prec.data(), d_res_float, num_elem*sizeof(float));
gpu_device.synchronize(); gpu_device.synchronize();