mirror of
				https://github.com/ml-explore/mlx.git
				synced 2025-10-31 07:58:14 +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
	 Cheng
					Cheng