mirror of
				https://github.com/ml-explore/mlx.git
				synced 2025-11-04 10:38:10 +08:00 
			
		
		
		
	The tile op does not accept None as reps (#1117)
This commit is contained in:
		@@ -3824,7 +3824,9 @@ void init_ops(nb::module_& m) {
 | 
			
		||||
    )pbdoc");
 | 
			
		||||
  m.def(
 | 
			
		||||
      "tile",
 | 
			
		||||
      [](const array& a, const IntOrVec& reps, StreamOrDevice s) {
 | 
			
		||||
      [](const array& a,
 | 
			
		||||
         const std::variant<int, std::vector<int>>& reps,
 | 
			
		||||
         StreamOrDevice s) {
 | 
			
		||||
        if (auto pv = std::get_if<int>(&reps); pv) {
 | 
			
		||||
          return tile(a, {*pv}, s);
 | 
			
		||||
        } else {
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user