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

Refined the testing of log and exp on fp16

Browse Source
This commit is contained in:
Benoit Steiner 2016-05-05 16:24:15 -07:00
parent 9a48688d37
commit e3d053e14e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/test/cxx11_tensor_of_float16_cuda.cu
View File

@ -159,8 +159,8 @@ void test_cuda_trancendental() {
Eigen::TensorMap<Eigen::Tensor<Eigen::half, 1>, Eigen::Aligned> gpu_res2_float( Eigen::TensorMap<Eigen::Tensor<Eigen::half, 1>, Eigen::Aligned> gpu_res2_float(
d_res2_float, num_elem); d_res2_float, num_elem);
gpu_float1.device(gpu_device) = gpu_float1.random(); gpu_float1.device(gpu_device) = gpu_float1.random() - 0.5f;
gpu_float2.device(gpu_device) = gpu_float2.random(); gpu_float2.device(gpu_device) = gpu_float2.random() + 0.5f;;
gpu_res1_float.device(gpu_device) = gpu_float1.exp().cast<Eigen::half>(); gpu_res1_float.device(gpu_device) = gpu_float1.exp().cast<Eigen::half>();
gpu_res2_float.device(gpu_device) = gpu_float2.log().cast<Eigen::half>(); gpu_res2_float.device(gpu_device) = gpu_float2.log().cast<Eigen::half>();
gpu_res1_half.device(gpu_device) = gpu_float1.cast<Eigen::half>().exp(); gpu_res1_half.device(gpu_device) = gpu_float1.cast<Eigen::half>().exp();