zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-28 03:41:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nit

Browse Source
This commit is contained in:
Hyunsung Lee 2025-04-22 05:55:41 +09:00
parent 5e8712cf6f
commit 605c6bc551
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/ops.cpp
View File

@ -5196,8 +5196,8 @@ void init_ops(nb::module_& m) {
throw std::invalid_argument( throw std::invalid_argument(
"[broadcast_shapes] Must provide at least one shape."); "[broadcast_shapes] Must provide at least one shape.");
mx::Shape result; mx::Shape result = nb::cast<mx::Shape>(shapes[0]);
for (size_t i = 0; i < shapes.size(); ++i) { for (size_t i = 1; i < shapes.size(); ++i) {
if (!nb::isinstance<mx::Shape>(shapes[i]) && if (!nb::isinstance<mx::Shape>(shapes[i]) &&
!nb::isinstance<nb::tuple>(shapes[i])) !nb::isinstance<nb::tuple>(shapes[i]))
throw std::invalid_argument( throw std::invalid_argument(