Commit 88de1cc9 authored by Yong Tang's avatar Yong Tang
Browse files

Add complex64 and complex128 support for `tf.angle` on GPU



In PR 10643, complex64 and complex128 support have been added
for `tf.angle` on CPU. However, because of the compilation
errors, the complex support on GPU is not enabled yet.

The issue was that, std::arg is not available on nvidia device
for GPU. This fix changes to used atan2 instead, which is
available in CUDA.

The relevant test cases have bee enabled.

This fix is related to 10643.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent a74f0484
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment