diff options
author | Rohit Santhanam <rohit.santhanam@amd.com> | 2021-04-11 14:54:18 +0000 |
---|---|---|
committer | David Tellenbach <david.tellenbach@me.com> | 2021-04-12 10:19:06 +0000 |
commit | dfd6720d823bf41fff9350fe7ff7c854903b54ad (patch) | |
tree | e3570896d07595bf5d85e923f304cdff6a8108ce /unsupported | |
parent | 1e1c8a735c96b2f630a6cefae09d57dd4b1dd4ed (diff) |
Fix for float16 GPU unit test.
Diffstat (limited to 'unsupported')
-rw-r--r-- | unsupported/test/cxx11_tensor_of_float16_gpu.cu | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/unsupported/test/cxx11_tensor_of_float16_gpu.cu b/unsupported/test/cxx11_tensor_of_float16_gpu.cu index c55676c76..062f76e26 100644 --- a/unsupported/test/cxx11_tensor_of_float16_gpu.cu +++ b/unsupported/test/cxx11_tensor_of_float16_gpu.cu @@ -444,7 +444,7 @@ void test_gpu_forced_evals() { d_float, num_elem); Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_res_half1( d_res_half1, num_elem); - Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Unaligned> gpu_res_half2( + Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Unaligned> gpu_res_half2( d_res_half2, num_elem); Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_res_float( d_res_float, num_elem); @@ -461,7 +461,7 @@ void test_gpu_forced_evals() { Tensor<float, 1> half_prec2(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_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.synchronize(); |