math.h File Reference

#include "mlx/backend/cpu/simd/type.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::simd

Functions
template<typename T, int N>
Simd< T, N >	mlx::core::simd::exp (Simd< T, N > in)
	Compute exp(x) in an optimizer friendly way as follows:
template<bool Sine, typename T, int N>
Simd< T, N >	mlx::core::simd::sincos (Simd< T, N > in)
template<typename T, int N>
Simd< T, N >	mlx::core::simd::sin (Simd< T, N > x)
template<typename T, int N>
Simd< T, N >	mlx::core::simd::cos (Simd< T, N > x)
template<typename T, int N>
Simd< T, N >	mlx::core::simd::erf (Simd< T, N > x)
template<typename T, int N>
Simd< T, N >	mlx::core::simd::erfinv (Simd< T, N > a_)

Variables
constexpr float	mlx::core::simd::inf = std::numeric_limits<float>::infinity()