zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-24 02:41:19 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats 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");
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 {