mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-24 02:41:19 +08:00
The tile op does not accept None as reps (#1117)
This commit is contained in:
parent
2c7e9b5158
commit
cbd5445ea7
@ -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 {
|
||||
|
Loading…
Reference in New Issue
Block a user