metal.h File Reference

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

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::metal

Functions
bool	mlx::core::metal::is_available ()
size_t	mlx::core::metal::get_active_memory ()
size_t	mlx::core::metal::get_peak_memory ()
void	mlx::core::metal::reset_peak_memory ()
size_t	mlx::core::metal::get_cache_memory ()
size_t	mlx::core::metal::set_memory_limit (size_t limit, bool relaxed=true)
size_t	mlx::core::metal::set_cache_limit (size_t limit)
void	mlx::core::metal::clear_cache ()
size_t	mlx::core::metal::set_wired_limit (size_t limit)
void	mlx::core::metal::start_capture (std::string path="")
	Capture a GPU trace, saving it to an absolute file path
void	mlx::core::metal::stop_capture ()
std::unordered_map< std::string, std::variant< std::string, size_t > >	mlx::core::metal::device_info ()
	Get information about the GPU and system settings.