Alex Barron 2e158cf6d0 Add conjugate operator (#1100) ... * cpu and gpu impl * add mx.conj and array.conj() --------- Co-authored-by: Alex Barron <abarron22@apple.com>		2024-05-10 07:22:20 -07:00
..
reduction	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
steel	Update block offset adjustment to be in size_t (#1087)	2024-05-08 08:10:23 -07:00
arange.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
arg_reduce.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
atomic.h	Fix eye for larger matrices (#463)	2024-01-16 00:51:24 -08:00
bf16_math.h	copyright + ack	2023-11-30 11:12:53 -08:00
bf16.h	copyright + ack	2023-11-30 11:12:53 -08:00
binary_two.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
binary.h	Added ArcTan2 operation (#1079)	2024-05-08 08:35:15 -07:00
binary.metal	Added ArcTan2 operation (#1079)	2024-05-08 08:35:15 -07:00
CMakeLists.txt	Metal FFT for powers of 2 up to 2048 (#915)	2024-04-11 21:40:06 -07:00
compiled_preamble.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
complex.h	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
conv.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
copy.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
defines.h	Fast RMS Norm (#862)	2024-03-21 07:20:54 -07:00
erf.h	copyright + ack	2023-11-30 11:12:53 -08:00
expm1f.h	std and expm1 (#973)	2024-04-08 14:26:01 -07:00
fft.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
gather.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
gemv.metal	Block sparse mm (#1058)	2024-05-02 14:03:58 -07:00
indexing.h	Update gather and scatter to not use Argument Encoder (#683)	2024-02-14 13:42:13 -08:00
layer_norm.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
quantized.metal	Improvements in the quantizer and dequantization kernel (#1061)	2024-05-01 18:19:11 -07:00
random.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
rms_norm.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
rope.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
scaled_dot_product_attention_params.h	Fast Inference SDPA op (#735)	2024-03-04 21:06:11 -08:00
scaled_dot_product_attention.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
scan.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
scatter.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
softmax.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
sort.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
ternary.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
ternary.metal	feat: metal formatting and pre-commit bump (#1038)	2024-04-30 07:18:09 -07:00
unary.h	Add conjugate operator (#1100)	2024-05-10 07:22:20 -07:00
unary.metal	Add conjugate operator (#1100)	2024-05-10 07:22:20 -07:00
utils.h	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -07:00