diff options
-rw-r--r-- | unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu b/unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu index 54c17ca28..2baf5eaad 100644 --- a/unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu +++ b/unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu @@ -15,7 +15,7 @@ #include <cuda_fp16.h> #endif #include "main.h" -#include <Eigen/CXX11/Tensor> +#include <unsupported/Eigen/CXX11/Tensor> using Eigen::Tensor; @@ -48,16 +48,16 @@ void test_cuda_complex_cwise_ops() { gpu_in2.device(gpu_device) = gpu_in2.constant(b); enum CwiseOp { - Add, + Add = 0, Sub, Mul, Div }; - Tensor<std::complex<T>, 1, 0, int> actual(2); - for (CwiseOp op : {Add, Sub, Mul, Div}) { + Tensor<std::complex<T>, 1, 0, int> actual(kNumItems); + for (int op = Add; op <= Div; op++) { std::complex<T> expected; - switch (op) { + switch (static_cast<CwiseOp>(op)) { case Add: gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; expected = a + b; |