diff options
| -rw-r--r-- | unsupported/test/cxx11_tensor_complex_cwise_ops_gpu.cu | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/unsupported/test/cxx11_tensor_complex_cwise_ops_gpu.cu b/unsupported/test/cxx11_tensor_complex_cwise_ops_gpu.cu index af67348aa..3b078d1d1 100644 --- a/unsupported/test/cxx11_tensor_complex_cwise_ops_gpu.cu +++ b/unsupported/test/cxx11_tensor_complex_cwise_ops_gpu.cu @@ -48,11 +48,13 @@ void test_cuda_complex_cwise_ops() { Add = 0, Sub, Mul, - Div + Div, + Neg, + NbOps }; Tensor<std::complex<T>, 1, 0, int> actual(kNumItems); - for (int op = Add; op <= Div; op++) { + for (int op = Add; op < NbOps; op++) { std::complex<T> expected; switch (static_cast<CwiseOp>(op)) { case Add: @@ -71,6 +73,10 @@ void test_cuda_complex_cwise_ops() { gpu_out.device(gpu_device) = gpu_in1 / gpu_in2; expected = a / b; break; + case Neg: + gpu_out.device(gpu_device) = -gpu_in1; + expected = -a; + break; } assert(cudaMemcpyAsync(actual.data(), d_out, complex_bytes, cudaMemcpyDeviceToHost, gpu_device.stream()) == cudaSuccess); |