Fix compilation with cuda 8

This commit is contained in:
Gael Guennebaud 2016-08-29 11:06:08 +02:00
parent 8f4b4ad5fb
commit 6cd7b9ea6b
2 changed files with 11 additions and 2 deletions

View File

@ -1,4 +1,11 @@
// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2015-2016 Gael Guennebaud <gael.guennebaud@inria.fr>
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
// workaround issue between gcc >= 4.7 and cuda 5.5
#if (defined __GNUC__) && (__GNUC__>4 || __GNUC_MINOR__>=7)
@ -12,6 +19,8 @@
#define EIGEN_DEFAULT_DENSE_INDEX_TYPE int
#include <math_constants.h>
#include <cuda.h>
#include <cuda_fp16.h>
#include "main.h"
#include "cuda_common.h"

View File

@ -13,7 +13,7 @@
#define EIGEN_DEFAULT_DENSE_INDEX_TYPE int
#define EIGEN_USE_GPU
#include <cuda_fp16.h>
#include "main.h"
#include <unsupported/Eigen/CXX11/Tensor>