zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
c4a471c99d
mlx/mlx/backend/common
History
Rifur13 c4a471c99d
Add groups to Conv1d (#948) 
* Add conv1d grouped convs on CPU

* Add GPU support

* Parallelize inside metal kernel

* clenaup

* Update mlx/ops.cpp

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* New unfold kernel + remove unused code

* Remove copy and refactor

* Update vjp and reuse steel gemm

* Fixed groups on cpu

* Fix metal validation

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-04-27 06:24:57 -07:00
..
arange.h copyright + ack 2023-11-30 11:12:53 -08:00
arg_reduce.cpp copyright + ack 2023-11-30 11:12:53 -08:00
binary_two.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
binary.cpp Add bitwise ops (#1037) 2024-04-26 22:03:42 -07:00
binary.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
CMakeLists.txt Masked mm (#978) 2024-04-16 14:45:39 -07:00
compiled_cpu.cpp Fix cpu compile (#934) 2024-04-01 17:37:12 -07:00
compiled_nocpu.cpp Ios compile (#784) 2024-03-04 20:02:26 -08:00
compiled_preamble.h Refactor CPU compile preamble (#708) 2024-02-19 06:12:53 -08:00
compiled.cpp Fix cpu compile (#934) 2024-04-01 17:37:12 -07:00
compiled.h Fix cpu compile (#934) 2024-04-01 17:37:12 -07:00
conv.cpp Add groups to Conv1d (#948) 2024-04-27 06:24:57 -07:00
copy.cpp Add missing && when forwarding args (#925) 2024-03-29 06:48:29 -07:00
copy.h Add a SliceUpdate op and primitive (#850) 2024-03-20 10:39:25 -07:00
default_primitives.cpp Masked mm (#978) 2024-04-16 14:45:39 -07:00
erf.cpp copyright + ack 2023-11-30 11:12:53 -08:00
fft.cpp copyright + ack 2023-11-30 11:12:53 -08:00
indexing.cpp copyright + ack 2023-11-30 11:12:53 -08:00
inverse.cpp Add vmap for SVD and inverse (#849) 2024-03-21 13:18:27 -07:00
lapack_helper.h Add SVD primitive (#809) 2024-03-12 12:30:11 -07:00
load.cpp Spelling (#342) 2024-01-01 21:08:17 -08:00
make_compiled_preamble.sh Refactor CPU compile preamble (#708) 2024-02-19 06:12:53 -08:00
masked_mm.cpp Masked mm (#978) 2024-04-16 14:45:39 -07:00
ops.h Add bitwise ops (#1037) 2024-04-26 22:03:42 -07:00
primitives.cpp std and expm1 (#973) 2024-04-08 14:26:01 -07:00
qrf.cpp fixes for building and running on iOS (#619) 2024-02-04 12:29:17 -08:00
quantized.cpp Custom VJP and checkpointing (#541) 2024-01-30 16:04:45 -08:00
reduce.cpp copyright + ack 2023-11-30 11:12:53 -08:00
reduce.h Implement vjps for some primitives in the fast namespace (#883) 2024-03-26 16:35:34 -07:00
scan.cpp add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
select.cpp Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
softmax.cpp Option for precise softmax (#953) 2024-04-04 08:32:35 -07:00
sort.cpp copyright + ack 2023-11-30 11:12:53 -08:00
svd.cpp Add vmap for SVD and inverse (#849) 2024-03-21 13:18:27 -07:00
ternary.h Implement the 'where' primitive for conditional selection (#664) 2024-02-22 15:10:48 -08:00
threefry.cpp copyright + ack 2023-11-30 11:12:53 -08:00
threefry.h copyright + ack 2023-11-30 11:12:53 -08:00
unary.h Refactor CPU compile preamble (#708) 2024-02-19 06:12:53 -08:00
utils.h Add missing && when forwarding args (#894) 2024-03-25 14:55:54 -07:00