Cheng 28fcd2b519 Add missing && when forwarding args (#894) ... Without the && args would be copied and perfect forwarding won't work. Also add template utils to make sure the function only forwards array and not vector<array>.		2024-03-25 14:55:54 -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 numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)	2024-03-25 12:32:59 -07:00
binary.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
CMakeLists.txt	Fast RMS Norm (#862)	2024-03-21 07:20:54 -07:00
compiled_cpu.cpp	Compile stride bug (#812)	2024-03-11 06:31:31 -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	Add types and order in kernel name (#831)	2024-03-13 20:34:06 -07:00
compiled.h	Shapeless compilation for some graphs (#687)	2024-02-19 21:43:54 -08:00
conv.cpp	Convolution update (#651)	2024-02-28 20:11:16 -08:00
copy.cpp	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -07:00
copy.h	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -07:00
default_primitives.cpp	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -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
ops.h	Fix logsumexp edge case (#740)	2024-02-25 08:39:55 -08:00
primitives.cpp	add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)	2024-03-25 12:32:59 -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	Refactor the reduction kernels (#277)	2023-12-24 14:47:57 -08: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	Fast RMS Norm (#862)	2024-03-21 07:20:54 -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