compiled.h File Reference

#include <iomanip>
#include <sstream>
#include <unordered_set>
#include "mlx/array.h"
#include "mlx/primitives.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
bool	mlx::core::is_static_cast (const Primitive &p)
std::string	mlx::core::build_lib_name (const std::vector< array > &inputs, const std::vector< array > &outputs, const std::vector< array > &tape, const std::unordered_set< uintptr_t > &constant_ids)
std::string	mlx::core::get_type_string (Dtype d)
template<typename T >
void	mlx::core::print_float_constant (std::ostream &os, const array &x)
template<typename T >
void	mlx::core::print_int_constant (std::ostream &os, const array &x)
template<typename T >
void	mlx::core::print_complex_constant (std::ostream &os, const array &x)
void	mlx::core::print_constant (std::ostream &os, const array &x)
bool	mlx::core::is_scalar (const array &x)
bool	mlx::core::compiled_check_contiguity (const std::vector< array > &inputs, const std::vector< int > &shape)
void	mlx::core::compiled_allocate_outputs (const std::vector< array > &inputs, std::vector< array > &outputs, const std::vector< array > &inputs_, const std::unordered_set< uintptr_t > &constant_ids_, bool contiguous, bool move_buffers=false)