diff options
author | RJ Ryan <rjryan@google.com> | 2016-09-20 07:49:05 -0700 |
---|---|---|
committer | RJ Ryan <rjryan@google.com> | 2016-09-20 07:49:05 -0700 |
commit | 608b1acd6d43bff23b3b136c546d88c939e4d37d (patch) | |
tree | b025123a5e9cc9fd612e85fb3b1da3f38bb91ef5 /unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu | |
parent | b2c6dc48d9189eb96f878aa6028aec245eadde85 (diff) |
Don't use c++11 features and fix include.
Diffstat (limited to 'unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu')
-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; |