L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add logicalAnd and logicalOR (#386)

Browse Source 
* feat: add logicalAnd and logicalOR

* run pre-commit

* Refactor logical_and and logical_or functions

* Add acknowledgement

* Add logical AND and logical OR operators

* Refactor logical_and and logical_or functions

* Add support for logical operators on bool arrays

* Update mlx/ops.cpp

Co-authored-by: Awni Hannun <[email protected]>

* Update mlx/ops.cpp

Co-authored-by: Awni Hannun <[email protected]>

* Add logical AND and OR operators for arrays and scalars

* Refactor vjp and jvp methods in primitives.cpp

* Add overloaded operators for logical AND and OR

* format

---------

Co-authored-by: Awni Hannun <[email protected]>
Co-authored-by: Awni Hannun <[email protected]>
This commit is contained in:
Nripesh Niketan
2024-01-08 07:00:05 -08:00
committed by GitHub
co-authored by Awni Hannun Awni Hannun
parent 022a944367
commit 73321b8097
17 changed files with 311 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/test_vmap.py
+2
View File
@@ -80,6 +80,8 @@ class TestVmap(mlx_tests.MLXTestCase):
"multiply",
"power",
"subtract",
"logical_or",
"logical_and",
]
for opname in ops:
with self.subTest(op=opname):