compile_impl.h File Reference

#include <unordered_map>
#include "mlx/array.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::detail

Typedefs
using	mlx::core::detail::ParentsMap

Functions
std::function< std::vector< array >(const std::vector< array > &)>	mlx::core::detail::compile (std::function< std::vector< array >(const std::vector< array > &)> fun, std::uintptr_t fun_id, bool shapeless=false, std::vector< uint64_t > constants={})
void	mlx::core::detail::compile_erase (std::uintptr_t fun_id)
void	mlx::core::detail::compile_clear_cache ()
bool	mlx::core::detail::compile_available_for_device (const Device &device)
std::pair< std::vector< array >, std::vector< array > >	mlx::core::detail::compile_trace (const std::function< std::vector< array >(const std::vector< array > &)> &fun, const std::vector< array > &inputs, bool shapeless)
std::pair< std::vector< array >, ParentsMap >	mlx::core::detail::compile_dfs (const std::vector< array > &inputs, const std::vector< array > &outputs, const std::vector< array > &original_inputs)
void	mlx::core::detail::compile_simplify (std::vector< array > &tape, ParentsMap &parents_map, std::vector< array > &outputs, int passes)
std::vector< array >	mlx::core::detail::compile_replace (const std::vector< array > &tape, const std::vector< array > &trace_inputs, const std::vector< array > &trace_outputs, const std::vector< array > &inputs, bool shapeless)
void	mlx::core::detail::compile_validate_shapeless (const std::vector< array > &tape)