MUDA
Loading...
Searching...
No Matches
eigen_cxx20.h
1
#pragma once
2
#include <muda/muda_def.h>
3
#ifdef __CUDA_ARCH__
4
#include <complex>
5
// Fix eigen cuda cxx20 : can't find `arg` in global scope
6
template
<
typename
T>
7
MUDA_INLINE MUDA_GENERIC T arg(
const
std::complex<T>& z)
8
{
9
return
std::atan2(std::imag(z), std::real(z));
10
}
11
#endif
src
muda
ext
eigen
eigen_cxx20.h
Generated by
1.9.8