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

More EIGEN_DEVICE_FUNC fixes for CUDA 10/11/12.

Browse Source
This commit is contained in:
Antonio Sánchez
2023-02-03 19:18:45 +00:00
committed by Rasmus Munk Larsen
parent 2a90653395
commit f6cc359e10
9 changed files with 68 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceGpu.h
View File

@@ -378,7 +378,7 @@ struct GpuDevice {
return stream_->deviceProperties().maxThreadsPerMultiProcessor;
}
EIGEN_STRONG_INLINE int sharedMemPerBlock() const {
return stream_->deviceProperties().sharedMemPerBlock;
return static_cast<int>(stream_->deviceProperties().sharedMemPerBlock);
}
EIGEN_STRONG_INLINE int majorDeviceVersion() const {
return stream_->deviceProperties().major;