mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-12-27 07:29:52 +08:00
070d303d56
This is to support scalar `sqrt` of complex numbers `std::complex<T>` on device, requested by Tensorflow folks. Technically `std::complex` is not supported by NVCC on device (though it is by clang), so the default `sqrt(std::complex<T>)` function only works on the host. Here we create an overload to add back the functionality. Also modified the CMake file to add `--relaxed-constexpr` (or equivalent) flag for NVCC to allow calling constexpr functions from device functions, and added support for specifying compute architecture for NVCC (was already available for clang). |
||
|---|---|---|
| .. | ||
| CXX11 | ||
| src | ||
| AdolcForward | ||
| AlignedVector3 | ||
| ArpackSupport | ||
| AutoDiff | ||
| BVH | ||
| CMakeLists.txt | ||
| EulerAngles | ||
| FFT | ||
| IterativeSolvers | ||
| KroneckerProduct | ||
| LevenbergMarquardt | ||
| MatrixFunctions | ||
| MoreVectorization | ||
| MPRealSupport | ||
| NonLinearOptimization | ||
| NumericalDiff | ||
| OpenGLSupport | ||
| Polynomials | ||
| Skyline | ||
| SparseExtra | ||
| SpecialFunctions | ||
| Splines | ||