compile.h File Reference

#include "mlx/array.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Enumerations
enum class	mlx::core::CompileMode { mlx::core::disabled , mlx::core::no_simplify , mlx::core::no_fuse , mlx::core::enabled }

Functions
std::function< std::vector< array >(const std::vector< array > &)>	mlx::core::compile (std::function< std::vector< array >(const std::vector< array > &)> fun, bool shapeless=false)
	Compile takes a function and returns a compiled function.
std::function< std::vector< array >(const std::vector< array > &)>	mlx::core::compile (std::vector< array >(*fun)(const std::vector< array > &), bool shapeless=false)
template<typename F>
std::function< std::vector< array >(const std::vector< array > &)>	mlx::core::compile (F &&f, bool shapeless=false)
void	mlx::core::disable_compile ()
	Globally disable compilation.
void	mlx::core::enable_compile ()
	Globally enable compilation.
void	mlx::core::set_compile_mode (CompileMode mode)
	Set the compiler mode to the given value.