devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made it possible to run tensor chipping operations on CUDA devices

Browse Source
This commit is contained in:
Benoit Steiner
2015-12-17 13:29:08 -08:00
parent 2ca55a3ae4
commit 40e6250fc3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/Eigen/CXX11/src/Tensor/TensorChipping.h
View File

@@ -50,7 +50,7 @@ struct nested<TensorChippingOp<DimId, XprType>, 1, typename eval<TensorChippingO
template <DenseIndex DimId> template <DenseIndex DimId>
struct DimensionId struct DimensionId
{ {
DimensionId(DenseIndex dim) { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DimensionId(DenseIndex dim) {
eigen_assert(dim == DimId); eigen_assert(dim == DimId);
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {
@@ -60,7 +60,7 @@ struct DimensionId
template <> template <>
struct DimensionId<Dynamic> struct DimensionId<Dynamic>
{ {
DimensionId(DenseIndex dim) : actual_dim(dim) { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DimensionId(DenseIndex dim) : actual_dim(dim) {
eigen_assert(dim >= 0); eigen_assert(dim >= 0);
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {