zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-31 07:58:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

The tile op does not accept None as reps (#1117)

Browse Source
This commit is contained in:
Cheng
2024-05-14 21:25:25 +08:00
committed by GitHub
parent 2c7e9b5158
commit cbd5445ea7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/ops.cpp
View File

@@ -3824,7 +3824,9 @@ void init_ops(nb::module_& m) {
)pbdoc"); )pbdoc");
m.def( m.def(
"tile", "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) { if (auto pv = std::get_if<int>(&reps); pv) {
return tile(a, {*pv}, s); return tile(a, {*pv}, s);
} else { } else {