From cbd5445ea740216f0cb3a98fbaba2683765cdb56 Mon Sep 17 00:00:00 2001 From: Cheng Date: Tue, 14 May 2024 21:25:25 +0800 Subject: [PATCH] The tile op does not accept None as reps (#1117) --- python/src/ops.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/src/ops.cpp b/python/src/ops.cpp index b468be0b0..009b1dace 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -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>& reps, + StreamOrDevice s) { if (auto pv = std::get_if(&reps); pv) { return tile(a, {*pv}, s); } else {