mlx/tests
Diogo dc2edc762c
added tri / tril / triu (#170)
* added tri / tril / triu

* fixed tests

* ctest tests

* tri overload and simplified tests

* changes from comment

* more tests for m

* ensure assert if not 2-D

* remove broadcast_to

* minor tweaks

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-15 17:30:34 -08:00
..
allocator_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
array_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
autograd_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
blas_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
CMakeLists.txt awni's commit files 2023-11-29 10:30:41 -08:00
creations_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
device_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
eval_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
fft_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
graph_optimize_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
load_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
metal_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
ops_tests.cpp added tri / tril / triu (#170) 2023-12-15 17:30:34 -08:00
random_tests.cpp random generation fix (#80) 2023-12-08 10:40:57 -08:00
scheduler_tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
tests.cpp copyright + ack 2023-11-30 11:12:53 -08:00
utils_tests.cpp Added mx.stack c++ frontend impl (#123) 2023-12-14 13:21:19 -08:00
vmap_tests.cpp Add move and swap axis, and vmap for slice, concat, and gather (#158) 2023-12-14 12:59:12 -08:00