mirror of
				https://github.com/ml-explore/mlx.git
				synced 2025-11-04 10:38:10 +08:00 
			
		
		
		
	Expose function to clear memory cache (#1032)
* expose function to clear memory cache * fix linux build * fix metal tests
This commit is contained in:
		@@ -90,6 +90,15 @@ void init_metal(nb::module_& m) {
 | 
			
		||||
      Returns:
 | 
			
		||||
        int: The previous cache limit in bytes.
 | 
			
		||||
      )pbdoc");
 | 
			
		||||
  metal.def(
 | 
			
		||||
      "clear_cache",
 | 
			
		||||
      &metal::clear_cache,
 | 
			
		||||
      R"pbdoc(
 | 
			
		||||
      Clear the memory cache.
 | 
			
		||||
 | 
			
		||||
      After calling this, :func:`get_cache_memory` should return ``0``.
 | 
			
		||||
      )pbdoc");
 | 
			
		||||
 | 
			
		||||
  metal.def(
 | 
			
		||||
      "start_capture",
 | 
			
		||||
      &metal::start_capture,
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user