mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-01-30 17:40:05 +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
Show Stats 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>
struct DimensionId
{
DimensionId(DenseIndex dim) {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DimensionId(DenseIndex dim) {
eigen_assert(dim == DimId);
}
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {
@ -60,7 +60,7 @@ struct DimensionId
template <>
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_DEVICE_FUNC EIGEN_STRONG_INLINE DenseIndex actualDim() const {