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

Don't use c++11 features and fix include.

Browse Source
This commit is contained in:
RJ Ryan 2016-09-20 07:49:05 -07:00
parent b2c6dc48d9
commit 608b1acd6d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
unsupported/test/cxx11_tensor_complex_cwise_ops_cuda.cu
View File

@ -15,7 +15,7 @@
#include <cuda_fp16.h> #include <cuda_fp16.h>
#endif #endif
#include "main.h" #include "main.h"
#include <Eigen/CXX11/Tensor> #include <unsupported/Eigen/CXX11/Tensor>
using Eigen::Tensor; using Eigen::Tensor;
@ -48,16 +48,16 @@ void test_cuda_complex_cwise_ops() {
gpu_in2.device(gpu_device) = gpu_in2.constant(b); gpu_in2.device(gpu_device) = gpu_in2.constant(b);
enum CwiseOp { enum CwiseOp {
Add, Add = 0,
Sub, Sub,
Mul, Mul,
Div Div
}; };
Tensor<std::complex<T>, 1, 0, int> actual(2); Tensor<std::complex<T>, 1, 0, int> actual(kNumItems);
for (CwiseOp op : {Add, Sub, Mul, Div}) { for (int op = Add; op <= Div; op++) {
std::complex<T> expected; std::complex<T> expected;
switch (op) { switch (static_cast<CwiseOp>(op)) {
case Add: case Add:
gpu_out.device(gpu_device) = gpu_in1 + gpu_in2; gpu_out.device(gpu_device) = gpu_in1 + gpu_in2;
expected = a + b; expected = a + b;