Angelos Katharopoulos 4eef8102c9 Distributed layers (#1270)

2025-03-21 13:52:17 -07:00

..

gemms

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

simd

fix simd erf_inv (#1896)

2025-02-24 13:57:47 -08:00

arange.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

arg_reduce.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

binary_ops.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

binary_two.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

binary.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

binary.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

cholesky.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

CMakeLists.txt

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

compiled_preamble.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

compiled.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

conv.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

copy.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

copy.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

distributed.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

eigh.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

encoder.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

encoder.h

Fix multistream GPU deadlock (#1969)

2025-03-20 07:19:47 -07:00

eval.cpp

Fix multistream GPU deadlock (#1969)

2025-03-20 07:19:47 -07:00

eval.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

fft.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

gemm.h

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

hadamard.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

indexing.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

inverse.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

jit_compiler.cpp

Fix compilation error on Windows (#1844)

2025-02-10 19:53:05 -08:00

jit_compiler.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

lapack.h

Double for lapack (#1904)

2025-02-25 11:39:36 -08:00

luf.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

make_compiled_preamble.ps1

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

make_compiled_preamble.sh

fix cpu compile (#1897)

2025-02-24 14:10:30 -08:00

masked_mm.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

matmul.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

primitives.cpp

Distributed layers (#1270)

2025-03-21 13:52:17 -07:00

qrf.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

quantized.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

reduce.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

scan.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

select.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

slicing.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

softmax.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

sort.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

svd.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

ternary.h

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

threefry.cpp

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

threefry.h

Refactor common into cpu specific and truly common (#1817)

2025-02-03 15:58:02 -08:00

unary_ops.h

Bitwise Inverse (#1862)

2025-02-13 08:44:14 -08:00

unary.cpp

reduce binary size (#1952)

2025-03-11 06:30:44 -07:00

unary.h

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00