docs/complex_8h.html
| | CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers |
Classes | Namespaces | Enumerations | Functions
complex.h File Reference
#include <cuComplex.h>
#include <cstdint>
#include "cutlass/cutlass.h"
#include "cutlass/half.h"
#include "cutlass/real.h"
#include <iosfwd>
Include dependency graph for complex.h:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
|
|
| class | cutlass::complex< T > |
| |
| struct | cutlass::RealType< complex< T > > |
| | Partial specialization for complex-valued type. More...
|
| |
|
| | | cutlass | | |
|
|
| enum | cutlass::ComplexTransform { cutlass::ComplexTransform::kNone, cutlass::ComplexTransform::kConjugate } |
| | Enumeraed type describing a transformation on a complex value. More...
|
| |
|
|
| CUTLASS_HOST_DEVICE float const & | cutlass::real (cuFloatComplex const &z) |
| | Returns the real part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE float & | cutlass::real (cuFloatComplex &z) |
| | Returns the real part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE double const & | cutlass::real (cuDoubleComplex const &z) |
| | Returns the real part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE double & | cutlass::real (cuDoubleComplex &z) |
| | Returns the real part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE float const & | cutlass::imag (cuFloatComplex const &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE float & | cutlass::imag (cuFloatComplex &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE double const & | cutlass::imag (cuDoubleComplex const &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| CUTLASS_HOST_DEVICE double & | cutlass::imag (cuDoubleComplex &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T const & | cutlass::real (complex< T > const &z) |
| | Returns the real part of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T & | cutlass::real (complex< T > &z) |
| | Returns the real part of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T const & | cutlass::imag (complex< T > const &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T & | cutlass::imag (complex< T > &z) |
| | Returns the imaginary part of the complex number. More...
|
| |
| template<typename T > |
| std::ostream & | cutlass::operator<< (std::ostream &out, complex< T > const &z) |
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T | cutlass::abs (complex< T > const &z) |
| | Returns the magnitude of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T | cutlass::arg (complex< T > const &z) |
| | Returns the magnitude of the complex number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE T | cutlass::norm (T const &z) |
| | Returns the squared magnitude of a real number. More...
|
| |
| template<> |
| CUTLASS_HOST_DEVICE int8_t | cutlass::norm (int8_t const &z) |
| | Returns the squared magnitude of a real number. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE double | cutlass::norm (complex< T > const &z) |
| | Returns the squared magnitude of a complex number. More...
|
| |
| template<typename T , typename R > |
| CUTLASS_HOST_DEVICE R | cutlass::norm_accumulate (T const &x, R const &accumulator) |
| | Norm-accumulate calculation. More...
|
| |
| template<typename T , typename R > |
| CUTLASS_HOST_DEVICE R | cutlass::norm_accumulate (complex< T > const &z, R const &accumulator) |
| | Norm accumulate specialized for complex types. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::conj (complex< T > const &z) |
| | Returns the complex conjugate. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::proj (complex< T > const &z) |
| | Projects the complex number z onto the Riemann sphere. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::polar (T const &r, T const &theta=T()) |
| | Returns a complex number with magnitude r and phase theta. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::exp (complex< T > const &z) |
| | Computes the complex exponential of z. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::log (complex< T > const &z) |
| | Computes the complex exponential of z. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::log10 (complex< T > const &z) |
| | Computes the complex exponential of z. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::sqrt (complex< T > const &z) |
| | Computes the square root of complex number z. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::cos (complex< T > const &z) |
| | Computes the cosine of complex z. More...
|
| |
| template<typename T > |
| CUTLASS_HOST_DEVICE complex< T > | cutlass::sin (complex< T > const &z) |
| | Computes the sin of complex z. More...
|
| |
| template<> |
| CUTLASS_HOST_DEVICE cutlass::complex< half_t > | cutlass::from_real< cutlass::complex< half_t > > (double r) |
| |
| template<> |
| CUTLASS_HOST_DEVICE cutlass::complex< float > | cutlass::from_real< cutlass::complex< float > > (double r) |
| |
| template<> |
| CUTLASS_HOST_DEVICE cutlass::complex< double > | cutlass::from_real< cutlass::complex< double > > (double r) |
| |
Generated by 1.8.11